python - 使用 Numpy 数组在数据帧中进行矢量化查找

Question

给定一个如下所示的 numpy 数组，是否可以使用向量化在不使用 for 循环的情况下返回 HHt 列中的关联值？

ex_arr = [2643, 2644, 2647]

for i in ex_arr:

    h_p = df.at[i, "HHt"]

示例 df:

<表类=“s-表”><标题>HHt <正文>2643126442264532646426475264862649726508

预期结果:125

Answer 1

使用DataFrame.loc ，如果需要列表或数组添加Series.to_list或Series.to_numpy :

print (df.loc[ex_arr,'HHt'].to_list())
[1, 2, 5]

print (df.loc[ex_arr,'HHt'].to_numpy())
[1 2 5]