快速搜索Pandas数据帧列

时间:2018-05-12 11:34:22

标签: python pandas dataframe spark-dataframe

我有一个Pandas数据框,其中一列包含字符串ID。我使用idxmax()来返回找到的ID的索引,但由于数据超过一百万行,因此搜索需要花费大量时间。有没有有效的搜索方式可以减少时间。我目前没有对ID进行排序。

1 个答案:

答案 0 :(得分:0)

请添加您正在处理的数据样本。

以下代码段将根据列对数据框进行排序,并选择第一行:

df.sort_values('column_name').iloc[0,0]