如何优化大熊猫groupby查找?

时间:2020-07-05 05:15:03

标签: pandas pandas-groupby

我有一个数据帧(15万行),其中我将4列分组,并希望找到其中三列与特定值匹配的行。

import pandas as pd
df = pd.read_csv('data.csv',encoding="ISO-8859-1",dtype='unicode')

g = df.groupby(['col1','col2','col3','col4'])
 
for i in g.groups:
    if (str(i[0]) == 'text1' and str(i[1]) == 'text2' and str(i[2]) == 'text3'):
        print (str(i[4]))

该代码大约需要2秒钟来进行查找。如何缩短执行时间?

0 个答案:

没有答案