我有一个数据帧(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秒钟来进行查找。如何缩短执行时间?