从excel列中选择一行，并根据Python中的另一列生成新列

Question

我是Python新手。我有和excel数据。我在Python中导入它。我想对它做些改动。例如，我有一些列，我想让它过滤我定义的一些列，根据这些行，它会给其他列写一个新的属性。只是为了更好地理解我有;

A       B      C 
1    red       -
3    white     -
2    white     -
3    pink      -
2    blue      -

例如，如果df.['A']=2和df.['B']=='blue'，它将在C列的对应行中写入'a1'，这样，我将以我想要的方式完成我的C列。我该怎么做呢?谢谢你的回答。

Answer 1

使用熊猫https://pandas.pydata.org/

import pandas as pd 
dataset = pd.read_excel('test.xlsx')
dataset.loc[(dataset['A'] == 2) & (dataset['B'] == 'blue') ,'C'] = 'a1'
dataset.to_excel('test1.xlsx')