对熊猫数据框使用if语句

时间:2020-11-10 06:51:23

标签: pandas dataframe if-statement

(底部代码)\

            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)

0 个答案:

没有答案