熊貓數據框的條件列輸出

Question

我有一個熊貓DataFrame看起來像這樣：

nameA     statusA     nameB     statusB
a         Q           x         X
b         Q           y         X
c         X           z         Q
d         X           o         Q
e         Q           p         X
f         Q           r         Q

我想打印基於以下規則此數據框的行：輸出列nameA如果statusA是Q否則，如果statusB是Q輸出列nameB 。 並且如果statusA和statusB均為Q ，則應同時輸出nameA和nameB列。

為此有一個內襯嗎？

更新：

預期輸出：

a,Q
b,Q
z,Q
o,Q
e,Q
f,Q,r,Q

Answer 1

> data['con'] = data['statusA'] + data['statusB']
> data.apply(lambda v: v['nameA'] if v['con'] == 'QX' else v['nameB'] if v['con'] == 'XQ' else v['nameA']+ ','+ v['nameB'], axis=1)
0     a
1     b
2     z
3     o
4     e
5    f,r
dtype: object

您可以使用字符串串聯產生確切的結果。

就像是

> data.apply(lambda v: v['nameA']+',Q' if v['con'] == 'QX' else v['nameB'] + ',Q' if v['con'] == 'XQ' else v['nameA']+ ',Q,' + v['nameB'] + ',Q', axis=1)
0        a,Q
1        b,Q
2        z,Q
3        o,Q
4        e,Q
5    f,Q,r,Q
dtype: object