配列にclassからの戻り値を格納したい

Question

import numpy as np
import pandas as pd
import gpxpy
import gpxpy.gpx
from pytz import timezone

from geopy.distance import geodesic

def calc_move_dist(Lat,Lng):
# データ数
N = Lat.shape[0]
# データ抽出
dist=[]

i1 = np.arange(0, N-1)
i2 = np.arange(1, N)

for j in range(N):
    x1=Lat[i1[j]]
    y1=Lng[i1[j]]
    x2=Lat[i2[j]]
    y2=Lng[i2[j]]
        # データマスク
    #mask = ((x1 != x2) & (y1 != y2))
    point1=(x1,y1)
    point2=(x2,y2)

    #dist=np.zeros(N-1)
        # データ間距離
    dist[j] = geodesic(point1,point2).km

    print(dist)
return dist

緯度経度から距離を求めるためgeodesic(point1,point2).km
からの戻り値をリストに格納し、return　でそのリストを返す
コードを書きたいの上のコードではdist[j]でエラーが出ます。
色々試みたのですが、解決できません。
dist[j]をdistにすると1つ1つの計算は出ます....
最終的には緯度経度のリストと、戻した距離のリストをcsvフ
ァイルに落とします。
おそらく根本的な事だと思いますがアドバイスお願いします。

Answer 1

i1とi2はnumpyのndarrayなので配列といってよいですが、distはリストです。

ndarrayは作成するときに要素数を決めます。
リストは、要素を増やすことは可能ですが、存在しないインデックスのところに代入することはできません。

これを解決するには、下のいずれかの方法を使います。
方法1　最初に必要な長さのリストを作っておく
方法2　appendを使って後ろに追加していく

C言語などに慣れている人は方法1の方が使いやすいでしょう。

dist=[] を以下に変更しましょう。
dist = [None] * N

そうすれば、そこでエラーがでることはなくなります。

Comments

[28taza]

有難うございました。
リストは存在しないインデックスに代入できない...
良く解りました。

Answer 2

distは要素数0のリストなので、要素に代入する操作dist[j] = ...に対してはエラーが出ます。