A Neighbor
0 5 1
1 4 0
2 1 3
3 2 2
4 10 0
你知道大熊猫的功能吗?由于我的数据集非常大,不幸的是太慢了。
谢谢!
答案 0 :(得分:1)
IDK,无论您是否检查我之前提问的更新答案,这都是我使用的方式
a=df.A.values[:,None]-df.A.values
np.fill_diagonal(a, 9999999)
np.argmin(abs(a),0)
Out[1160]: array([1, 0, 3, 2, 0], dtype=int64)
答案 1 :(得分:1)
这样的事情可能会起到作用:
df['myindex'] = df.index
# result = df.sort(['A'])
result = df.sort_values(by='A')
result['NeighborIndex'] = result['myindex'].shift(-1)