来自两个系列的笛卡尔积,长度和索引不同

时间:2018-07-18 12:17:11

标签: python arrays pandas numpy dataframe

给出两个系列:

import pandas as pd
ser1 = pd.Series(data = [1,2,3], index=[1,2,3])
ser2 = pd.Series(data = [1,2,3,4,5], index = ['a','b','c','d','e'])

如何将两者相乘以获得所需的输出?

pd.DataFrame(
data = [[1,2,3],[2,4,6],[3,6,9],[4,8,12],[5,10,15]],
index = ser2.index,
columns = ser1.index,)

到目前为止,我的方法是使用索引和cols匹配两个系列的临时数据帧,然后使用iteritems()遍历两个系列之一。我觉得应该有一种更清洁的方法来实现这一目标。

2 个答案:

答案 0 :(得分:4)

我认为两个Series的外部乘积需要numpy.outer

df = pd.DataFrame(np.outer(ser2, ser1), index = ser2.index, columns = ser1.index)

print (df)
   1   2   3
a  1   2   3
b  2   4   6
c  3   6   9
d  4   8  12
e  5  10  15

答案 1 :(得分:4)

您可以使用NumPy进行广播,以将一个系列的值乘以另一个系列的转换后的值。

res = pd.DataFrame(ser1.values * ser2.values[:, None],
                   index=ser2.index, columns=ser1.index)

print(res)

   1   2   3
a  1   2   3
b  2   4   6
c  3   6   9
d  4   8  12
e  5  10  15