Saya memiliki matriks jarak yang disimpan sebagai pandas dataframe df yang berisi jarak antara 400 pasang koordinat dalam kilometer sebagai berikut: 0 1 2 3 4 ...

1
Macter 5 April 2021, 23:44

2 jawaban

Jawaban Terbaik

Anda dapat menggunakan concat dan ide yang sama yang Anda gunakan dengan min tetapi dengan idxmin.

m = df>0
res = pd.concat([df[m].min(axis=1), df[m].idxmin(axis=1)], 
                axis=1, keys=['nearest_dist','nearest_id'])

print(res)
   nearest_dist nearest_id
0      1.698938          4
1      1.261969          4
2      0.420828          3
3      0.420828          2
4      0.812797          2

Atau bahkan lebih sederhana dengan agg

res = (df[m].agg(['min', 'idxmin'], axis=1)
            .rename(columns={'min':'nearest_dist', 'idxmin':'nearest_id'}))
3
Ben.T 5 April 2021, 20:56

stack lalu groupby + idxmin untuk mengiris Seri.

s = df.rename_axis(columns='nearest_id').stack().loc[lambda x: x > 0]
s = (s.loc[s.groupby(level=0).idxmin()]
      .to_frame('nearest_dist')
      .reset_index(-1))

  nearest_id  nearest_dist
0          4      1.698938
1          4      1.261969
2          3      0.420828
3          2      0.420828
4          2      0.812797

Dapat juga membangun DataFrame dari numpy.

arr = df.where(df > 0).to_numpy()

pd.DataFrame({'nearest_id': np.nanargmin(arr, 1), 
              'nearest_dist': np.nanmin(arr, 1)},
             index=df.index)
3
ALollz 5 April 2021, 21:06