代码之家 › 专栏 › 技术社区 › Riccardo Bucco

将单个数据帧行分解为多个数据帧行

dataframe pandas python-3.x python

Riccardo Bucco · 技术社区 · 3 年前

我的数据帧有一些列,其中每个值可以是“1”、“2”、“3”或“任意”。举个例子:

>>> df = pd.DataFrame({'a': ['1', '2', 'any', '3'], 'b': ['any', 'any', '3', '1']})
>>> df
     a    b
0    1  any
1    2  any
2  any    3
3    3    1

在我的例子中,“any”表示值可以是“1”、“2”或“3”。我想只使用值“1”、“2”和“3”(或者,通常,我可能有的任何值列表)来生成所有可能的行。下面是上述示例的预期输出:

我用这种丑陋而复杂的方法得到了这个结果:

a = df['a'].replace('any', '1,2,3').apply(lambda x: eval(f'[{str(x)}]')).explode()
result = pd.merge(df.drop(columns=['a']), a, left_index=True, right_index=True)
b = result['b'].replace('any', '1,2,3').apply(lambda x: eval(f'[{str(x)}]')).explode()
result = pd.merge(result.drop(columns=['b']), b, left_index=True, right_index=True)
result = result.drop_duplicates().reset_index(drop=True)

1 回复 | 直到 3 年前

Quang Hoang 3 年前

可以替换字符串 any 例如。 '1,2,3'

(df.replace('any', '1,2,3')
   .apply(lambda x: x.str.split(',') if x.name in ['a','b'] else x)
   .explode('a').explode('b')
   .drop_duplicates(['a','b'])
)

输出:

Eelco van Vliet 3 年前

我不会使用eval和字符串操作,而只是用一组值替换任何值

df[df == 'any'] = {1, 2, 3}
for col in df:
    df = df.explode(col)
df = df.astype(int).drop_duplicates()

推荐文章

user1245262 · 筛选Pandas数据帧时出现问题

1 年前

Foroand · 熊猫数据帧中的词频计数耗时过长

1 年前

user14696236 · 如何为每个对应的列创建一行[重复]

2 年前

Shawn Hemelstrand · 为什么我的自定义errorbar函数不能在R中工作?

2 年前

Karim Abou El Naga · 将带字符串的DataFrame绘制到堆叠条形图中

2 年前

The Great · 拆分并存储数据帧,但名称基于特定列中的唯一值

2 年前

nickolakis · 基于R中的列名复制列

2 年前

opposity · 形成一个数据帧,该数据帧包含R中包含类别和子类别的列

2 年前

A. Handler · 有没有办法将数据帧的列与完整列名向量相匹配?

2 年前

JasonX · 运行减法计算

2 年前