用于pos数据的python中的数据清理for循环

Question

我有邮件商店的POS数据。 数据如附图所示。

##read data from csv
data = pd.read_csv('test1.csv')
#make a kist for each column
sales_id = list(data['sales_id'])
shop_number = list(data['shop_number'])
sales = list(data['sales'])
cashier_no = list(data['cashier_no'])
messager_no = list(data['messager_no'])
type_of_sale = list(data['type_of_sale'])
costomer_ID = list(data['costomer_ID'])
type_of_sale = list(data['type_of_sale'])
date = list(data['date'])
time = list(data['time'])

我要创建一个新列表，显示应该删除购买数据。 像这样：

data_to_clean= [0,1,0,1,0,0,1,0,1]

要做到这一点，我想做一个for循环

for i in range(len(type_of_sale)):
    data_to_clean=[]
    if type_of_sale[i] == "purchase":
        data_to_clean = data_to_clean.append(0)
    elif type_of_sale[i] == "return":
        data_to_clean = data_to_clean.append(1)
        ## I want to write a code so I can delete purchasse data too 
        #with conditions if it has the same shop_number,messager_no,costomer_ID and -price

    return list(data_to_clean)

这段代码有两个主要问题。 一不动。 第二，我不知道如何检查shop_number ， messager_no和costomer_ID把1或0在我data_to_clean名单。 有时我要检查的数据等以上sales_id(1628060)有时它的下面等sales_id(1599414)即使知道出纳员可以不同。 但constomer_Id应该始终相同。

问题是如何编写代码，这样我可以创建一个列表或带有0和1的数据框，以显示应删除的数据。

Answer 1

当您想将数据与Python中的字符串进行比较时，应将此string放在qoutes中：

for i in range(len(type_of_sale)):
        data_to_clean=[]
        if type_of_sale[i] == "purchase": # here
            data_to_clean = data_to_clean.append(0)
        elif type_of_sale[i] == "return": # and here
            data_to_clean = data_to_clean.append(1)

Answer 2

检查熊猫文档 。 获取作为退货订单的物品可以很简单

returns = data.loc[data['type_of_sale'] == 'return']

如果要销售收银员90

data.loc[(data['type_of_sale'] == 'purchase') & (data['cashier_no'] == 90)]