python pandas：计算矩阵subs中列的argmax

Question

考虑一下玩具数据帧df1和df2，其中df2是df1的子集（不包括第一行）。在

将熊猫作为pd导入 将numpy作为np导入

df1 = pd.DataFrame({'colA':[3.0,9,45,7],'colB':['A','B','C','D']})
df2 = df1[1:]

现在让我们找到每帧的可乐

^{pr2}$

如果我的insterest矩阵是df2，我如何解决这个索引问题？这种怪癖是与熊猫、纽比有关，还是仅仅与Python的记忆有关？在

Answer 1

我想是因为索引。分配^{时，可以使用^{}：

df1 = pd.DataFrame({'colA':[3.0,9,45,7],'colB':['A','B','C','D']})
df2 = df1[1:].reset_index(drop=True)

In [464]: np.argmax(df1.colA)
Out[464]: 2

In [465]: np.argmax(df2.colA)
Out[465]: 1

我认为使用方法^{}代替np.argmax：

^{pr2}$

Answer 2

您需要重置df2的索引：

df2.reset_index(inplace=True, drop=True)
np.argmax(df2.colA)
>> 1