(底部代码)\
price_a price_b companies
2019-01-03 25 50 A
2019-03-03 30 60 B
2019-03-19 23 46 A
让我们说我上面的数据框名为“ df”。 我想在“公司”列下提取公司“ A”的整行。 我在想如何实现if语句:
if df['companies'] == 'A':\
print.....
CODE for df:
p = [25,30,23]
dates = ['2019-01-03', '2019-03-03', '2019-03-19']
df = pd.DataFrame(p, index = dates, columns = ['price_a'])
df['price_b'] = df['price_a'] * 2
df.loc['2019-01-03', 'companies'] = 'A'
df.loc['2019-03-03', 'companies'] = 'B'
df.loc['2019-03-19', 'companies'] = 'A'
print(df)