给定一个如下所示的Numy数组,是否可以在不使用for循环的情况下使用矢量化来返回列HHT中的关联值?
ex_arr = [2643, 2644, 2647]
for i in ex_arr:
h_p = df.at[i, "HHt"]
示例df:
HHt | |
---|---|
2643 | 1 |
2644 | 2 |
2645 | 3 |
2646 | 4 |
2647 | 5 |
2648 | 6 |
2649 | 7 |
2650 | 8 |
预期结果: 1 2 5
给定一个如下所示的Numy数组,是否可以在不使用for循环的情况下使用矢量化来返回列HHT中的关联值?
ex_arr = [2643, 2644, 2647]
for i in ex_arr:
h_p = df.at[i, "HHt"]
示例df:
HHt | |
---|---|
2643 | 1 |
2644 | 2 |
2645 | 3 |
2646 | 4 |
2647 | 5 |
2648 | 6 |
2649 | 7 |
2650 | 8 |
预期结果: 1 2 5
使用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]