Python3熊猫 - 如何创建4列的所有组合，并将它们写入CSV中的行？

Question

我有一列有4个CSV文件。每一列代表一个名称的一部分（4份）：

CSV 1：

first_name 
michael 
madonna 
steve 
albert 

CSV 2：

second_name 
luke 
han 
kurt 

CSV 3：

first_last_name 
jackson 
jobs 
skywalker 

CSV 4：

second_last_name 
solo 
cobain 
einstein 

最终的结果我想是所有的4列（4个CSV的）之间，以获得所有可能的组合：

first_name,second_name,first_last_name,second_last_name 
michael,luke,jackson,solo 
michael,luke,jackson,cobain 
michael,luke,jackson,einstein 
michael,luke,jobs,solo 
michael,luke,jobs,cobain 
michael,luke,jobs,einstein 
michael,luke,skywalker,solo 
michael,luke,skywalker,cobain 
michael,luke,skywalker,einstein 
... 

随着熊猫我转换每个CSV纳入数据框但我不知道如何将所有四个组合。我怎样才能做到这一点？

Answer 1

使用itertools.product进行繁重的工作。

import pandas as pd 
from itertools import product 

lists = [list(pd.read_csv('data{}.csv'.format(i), header=0).iloc[:,0]) for i in range(1,5)] 
combined = list(','.join(items) for items in product(*lists)) 
pd.DataFrame(combined).to_csv('combined.csv', index=0) 

如果您只是想要列表形式，请使用combined。它看起来像：

['michael,luke,jackson,solo', 
'michael,luke,jackson,cobain', 
'michael,luke,jackson,einstein', 
'michael,luke,jobs,solo', 
'michael,luke,jobs,cobain', 
'michael,luke,jobs,einstein', 
'michael,luke,skywalker,solo', 
'michael,luke,skywalker,cobain', 
'michael,luke,skywalker,einstein', 
... 

或者最后一行将组合值写入CSV。

Answer 2

import numpy as np 
import pandas as pd 
import itertools 
import functools 

def cartesian(df1, df2): 
    rows = itertools.product(df1.iterrows(), df2.iterrows())  
    df = pd.DataFrame(left.append(right) for (_, left), (_, right) in rows) 
    return df.reset_index(drop=True) 

df1 = pd.read_csv('first_name.csv') 
df2 = pd.read_csv('second_name.csv') 
df3 = pd.read_csv('first_last_name.csv') 
df4 = pd.read_csv('second_last_name.csv') 

combined = functools.reduce(cartesian, [df1, df2, df3, df4]) 
combined.to_csv('combined.csv')