pandas 向多级列 Dataframe 添加条件列

fzwojiic 于 2023-06-20 发布在其他

关注(0)|答案(2)|浏览(118)

我有一个多级列数据框架，我需要根据其他列中的条件添加一个列（level_2）。添加的列将应用于所有level_1组。例如，条件列将被添加到A1、B1、C1等。这个dataframe只是我正在研究的更大的dataframe的一个例子。Level_1是动态的并且可以改变，例如，添加Z1或取出B1。

import pandas as pd
import numpy as np

level_1 = ['A1', 'A1', 'A1', 'B1', 'B1', 'B1', 'C1', 'C1', 'C1']
level_2 = ['a_1', 'b_1', 'c_1', 'a_1', 'b_1', 'c_1', 'a_1', 'b_1', 'c_1']
data = [['a', 23, 'h', 'o', 45, 'v', 'a3', 1, 'b1'], ['b', 34, 'i', 'p', 3, 'w', 'a4', 32, 'b2'], ['c', 5, 'j', 'q', 7, 'x', 'a5', 6, 'b3'], ['d', 2, 'k', 'r', 5, 'y', 'a6', 76, 'b4'], ['e', 78, 'l', 's', 65, 'z', 'a7', 9, 'b5'], ['f', 98, 'm', 't', 23, 'a1',  'a8', 14, 'b6'], ['g', 3, 'n', 'u', 1, 'a2', 'a9', 45, 'b7']]
columns = pd.MultiIndex.from_tuples(list(zip(level_1, level_2)))
df1 = pd.DataFrame(data, columns=columns)
date = ['1/1/2023','1/2/2023','1/3/2023','1/4/2023','1/5/2023','1/6/2023','1/7/2023']

df1.insert(0, 'date', date)

df1.set_index('date', inplace=True)

我已经尝试了下面的代码，它工作，但我想知道是否有一个更有效的方法来做到这一点，没有循环？谢谢你。

for column_name in df1.columns.get_level_values(0).unique():
    df1.loc[(df1[column_name, 'b_1'] > 30) | (df1[column_name, 'a_1'] == 'c'), (column_name,'e_1')] = 1

df1 = df1.reindex(columns=['A1','B1','C1'], level=0)

pandas

来源：https://stackoverflow.com/questions/76484301/adding-conditional-columns-to-multi-level-column-dataframe

2条答案

按热度按时间

wdebmtf21#

有一个更简单的方法来重塑。堆叠level=0列值，然后根据所需条件分配新列e_1，最后反堆叠以重新整形回原始形式

s = df1.stack(level=0)
s.loc[s['a_1'].eq('e') & s['b_1'].gt(30), 'e_1'] = 1
s = s.unstack().swaplevel(axis=1).sort_index(axis=1)

A1               B1              C1            
         a_1 b_1 c_1  e_1 a_1 b_1 c_1 e_1 a_1 b_1 c_1 e_1
date                                                     
1/1/2023   a  23   h  NaN   o  45   v NaN  a3   1  b1 NaN
1/2/2023   b  34   i  NaN   p   3   w NaN  a4  32  b2 NaN
1/3/2023   c   5   j  NaN   q   7   x NaN  a5   6  b3 NaN
1/4/2023   d   2   k  NaN   r   5   y NaN  a6  76  b4 NaN
1/5/2023   e  78   l  1.0   s  65   z NaN  a7   9  b5 NaN
1/6/2023   f  98   m  NaN   t  23  a1 NaN  a8  14  b6 NaN
1/7/2023   g   3   n  NaN   u   1  a2 NaN  a9  45  b7 NaN

赞(0）回复(0）举报 2023-06-20

gopyfrb32#

可以使用.xs吗？

x = df1.xs('b_1', axis=1, level=1) > 30
y = df1.xs('a_1', axis=1, level=1).eq('c')
z = (x | y).astype(int)
z.columns = pd.MultiIndex.from_product([z.columns, ['e_1']])

df1 = pd.concat([df1, z], axis=1).reindex(columns=['A1','B1','C1'], level=0)
print(df1)

图纸：

A1              B1              C1            
         a_1 b_1 c_1 e_1 a_1 b_1 c_1 e_1 a_1 b_1 c_1 e_1
date                                                    
1/1/2023   a  23   h   0   o  45   v   1  a3   1  b1   0
1/2/2023   b  34   i   1   p   3   w   0  a4  32  b2   1
1/3/2023   c   5   j   1   q   7   x   0  a5   6  b3   0
1/4/2023   d   2   k   0   r   5   y   0  a6  76  b4   1
1/5/2023   e  78   l   1   s  65   z   1  a7   9  b5   0
1/6/2023   f  98   m   1   t  23  a1   0  a8  14  b6   0
1/7/2023   g   3   n   0   u   1  a2   0  a9  45  b7   1

赞(0）回复(0）举报 2023-06-20

我来回答

pandas 向多级列 Dataframe 添加条件列

2条答案

相关问题

热门标签

最新问答