pandas.DataFrameでの行方向の各要素に対して複数要素の計算結果を持つ列を追加したい

Question

pandasでの加工方法について教えて下さい。

下のようなDataFrameがあるとします。

このA列の要素に対して[-2,-1]それぞれをした結果を得たいと考えています。
（結果イメージは下記。得たいのはA・B・D列を含むDataFrame。C列は加算対象の[-2,-1]を表示したもので無くても良い。）

A列の各要素に対して2行ずつの結果となるので4行に増えた形でDataFrameを得たいです。

まずは、A,B,C列のDataFrameを作成しようとして、
A・BのDataFrameから1行ずつ取得して、forループで[-2,-1]を計算してconcatで積み上げたり、
A・BのDataFrameとCのSeriesを外部結合でmergeするような形で試したのですが、
思ったような形で作成できず、質問させて頂きました。

初歩的な質問で恐縮ですが、教えて頂けると幸いです。

Answer 1

A・BのDataFrameとCのSeriesを外部結合でmergeするような形で試したのですが、

how='cross'を使うとできます。
参考：cartesian product in pandas

import pandas as pd

df1 = pd.DataFrame({'A':[10,30], 'B':[20,40]})
df2 = pd.DataFrame({'C':[-2,-1]})

df = pd.merge(df1, df2, how='cross')
df['D'] = df['A'] + df['C']
print(df)
#    A   B  C   D
#0  10  20 -2   8
#1  10  20 -1   9
#2  30  40 -2  28
#3  30  40 -1  29

Comments

[mana40]

出来ました。ありがとうございました！