熊猫将多列相乘以创建新的df

时间:2018-09-03 21:41:41

标签: python pandas

我从两个数据帧开始-一个在一周中的一天填充虚拟变量,另一个在一天中的某个小时填充虚拟变量(HE-小时结束)。

df1 =一周中的天数= n x m,其中n是观测次数,m为6:

    Mon Tue Wed Thu Fri Sat
0   1   0   0   0   0   0
1   0   1   0   0   0   0
2   0   0   1   0   0   0
3   0   0   0   1   0   0

df2 =一天中的小时= n x l,其中n是整数,m是23:

    HE1 HE2 HE3 HE4 HE5 ... HE22 HE23
0   1   0   0   0   0   ... 0    0
1   0   1   0   0   0   ... 0    0
2   0   0   1   0   0   ... 0    0
3   0   0   0   1   0   ... 0    0

我想创建一个尺寸为n x(m乘以l)的新df,其中m乘以l = 6 * 23 = 138:

    MonHE1 MonHE2 MonHE3 MonHE4 MonHE5 ... SatHE22 SatHE23
0   1      0      0      0      0      ... 0       0
1   0      1      0      0      0      ... 0       0
2   0      0      1      0      0      ... 0       0
3   0      0      0      1      0      ... 0       0

我成功了:

hoursXdays = pd.DataFrame()
Mon = hours.multiply(days['Mon'],axis='index').add_prefix('Mon')
Tue = hours.multiply(days['Tue'],axis='index').add_prefix('Tue')
Wed = hours.multiply(days['Wed'],axis='index').add_prefix('Wed')
Thu = hours.multiply(days['Thu'],axis='index').add_prefix('Thu')
Fri = hours.multiply(days['Fri'],axis='index').add_prefix('Fri')
Sat = hours.multiply(days['Sat'],axis='index').add_prefix('Sat')
hoursXdays = pd.concat([Mon, Tue, Wed, Thu, Fri, Sat],axis=1)

我将需要这样做,而不仅仅是少数几个。如果有什么可以概括的方法,我将不胜感激。

基本上,大熊猫中是否有办法将两个维度为n x m,n x l的数据帧相乘,以生成维度为n x(m乘以l)的新df?

2 个答案:

答案 0 :(得分:3)

您可以在此处使用for循环

pd.concat([hours.multiply(days[x],axis='index').add_prefix(x) for x in days.columns],1)

答案 1 :(得分:1)

您可以使用itertools.productpd.DataFrame默认构造函数

d = {a[0]+b[0]: np.array(a[1])*b[1] for a,b in itertools.product(df.to_dict('list').items(), df2.to_dict('list').items())}

dff = pd.DataFrame(d)

MonHE1  MonHE2  ... SatHE22 SatHE23
0   1   0       ... 0       0
1   0   0       ... 0       0
2   0   0       ... 0       0
3   0   0       ... 0       0