以以下數(shù)據(jù)框?yàn)槔篸f = pd.DataFrame({'Sample':['X', 'Y', 'Z'], 'Base':[2, 10, 3], 'A':[0,5,100], 'C':[0,10,7]})我想添加一個名為df["indices"]列索引的新列df["A"]和/或df["C"]只要它們滿足兩個條件：必須大于 5df["A"]/df["Base"]或者df["C"]/df["Base"]必須大于或等于 1生成的數(shù)據(jù)框?qū)⑹牵篸f = pd.DataFrame({'Sample':['X', 'Y', 'Z'], 'Base':[2, 20, 3], 'A':[0,6,100], 'C':[0,10,7], 'indices': ['','C','A,C']})我可以為我的第一個條件獲取 True 或 False 值，df[['A','C']] > 5但我無法讓它與我的條件 2 一起使用，該條件基于我的數(shù)據(jù)框中的另一列。在新專欄中獲取 True 的索引又是另一回事了。我想象有 apply 和 get_loc 或 index 的東西，但無論我如何嘗試，我都無法讓它工作。
2 回答
元芳怎么了
TA貢獻(xiàn)1798條經(jīng)驗(yàn) 獲得超7個贊
讓我們創(chuàng)建一個滿足兩個給定條件的布爾掩碼，然后使用DataFrame.dot該掩碼來獲取索引：
m = df[['A', 'C']].gt(5) & df[['A', 'C']].div(df['Base'], axis=0).ge(1)
df['indices'] = m.dot(m.columns + ',').str.rstrip(',')
Sample Base A C indices
0 X 2 0 0
1 Y 10 5 10 C
2 Z 3 100 7 A,C
慕絲7291255
TA貢獻(xiàn)1859條經(jīng)驗(yàn) 獲得超6個贊
當(dāng)滿足任意數(shù)量的條件時，您可以將df.loc值分配回該列。一個簡單的方法是擁有其中 3 個，每個都有您想要的條件。如果您愿意，您也可以將它們串聯(lián)起來np.where以實(shí)現(xiàn)相同的目標(biāo)。
import pandas as pd
df = pd.DataFrame({'Sample':['X', 'Y', 'Z'],
'Base':[2, 10, 3],
'A':[0,5,100],
'C':[0,10,7]})
df.loc[(df['A'] / df['Base'] >=1) & (df['C'] / df['Base'] >=1), 'indicies'] = 'A,C'
df.loc[(df['A'] / df['Base'] >=1) & (df['C'] / df['Base'] <1), 'indicies'] = 'A'
df.loc[(df['A'] / df['Base'] <1) & (df['C'] / df['Base'] >=1), 'indicies'] = 'C'
輸出
Sample Base A C indicies
0 X 2 0 0 NaN
1 Y 10 5 10 C
2 Z 3 100 7 A,C
添加回答
舉報
0/150
提交
取消