如何从数据集中提取特定行并附加到另一个列表中

Question

我有一个数据集。大商场的销售。首先，我选择列（ITEM_TYPE），然后使用for循环匹配整个列中的（Dairy）。我面临着一个问题，我无法理解整排人。i、 我是初学者 Dataset

mainlist
df = pd.read_csv('bm_Train.csv')
dt =df['Item_Type']
list_of_item = dt.to_list()

i=1
for x in list_of_item:
    if x=="Dairy":
        mainlist.append(dt[:i])
i+=i

mainlist

Answer 1

如果要基于'Item_Type' == 'Dairy'筛选数据集，可以尝试使用以下代码

df_new = df.loc[df['Item_Type'] == 'Dairy'].copy()

另一方面，如果您只需要'Item_Type'的值

list_of_item = df['Item_Type'].unique().tolist()

否则，请告诉我你想要什么

Answer 2

如果要将符合筛选条件的行的内容提取到列表中，可以使用.loc筛选内容，然后使用.values.tolist()将行内容转换到列表中，如下所示：

mainlist = df.loc[df['Item_Type'] == "Dairy"].values.tolist()