如何检索其列(1)值连续等于列表值的Pandas DataFrame行?
例如,给定以下内容:
import pandas as pd
df = pd.DataFrame({'col1': [10, 20, 30, 40, 50, 88, 99, 30, 40, 50]})
lst = [30, 40, 50]
我想提取从30到50的数据帧行,但只提取第一个连续的值序列(只提取2到4个索引行).
如何检索其列(1)值连续等于列表值的Pandas DataFrame行?
例如,给定以下内容:
import pandas as pd
df = pd.DataFrame({'col1': [10, 20, 30, 40, 50, 88, 99, 30, 40, 50]})
lst = [30, 40, 50]
我想提取从30到50的数据帧行,但只提取第一个连续的值序列(只提取2到4个索引行).
这应该会起到作用:
df = pd.DataFrame({'col1': [10, 20, 30, 40, 50, 88, 99, 30, 40, 50]})
lst = [30, 40, 50]
ans=[]
for i,num in enumerate(df['col1']):
if num in lst:
lst.remove(num)
ans.append(i)
print(ans)