每个行是代表客户和代表该客户的一些特征的每一列。
我想获得行号并将它们添加到列表中,或根据其特征值将它们添加到列表中。
比方说,我们希望找到的行数,如果FEATUREA小于100或FEATUREB超过500
我已经写了一些代码,这是你可以看到下面。
import pandas as pd
d = [{'feature1': 100, 'feature2': 520, 'feature3': 54},
{'feature1': 102, 'feature2': 504, 'feature3': 51},
{'feature1': 241, 'feature2': 124, 'feature3': 4},
{'feature1': 340, 'feature2': 830, 'feature3': 700},
{'feature1': 98, 'feature2': 430, 'feature3': 123}]
df = DataFrame(d)
print(df)
print("----")
dataframe1 = df[(df['feature1'] < 100)]
dataframe2 = df[(df['feature2'] > 500)]
print(dataframe1)
print(dataframe2)
# he