Pandas merge 結合キーについて

Question

下記SQLのように結合キーを+1でデータフレームを作成したいのですが、
pandasの場合どのようにしたらいいでしょうか。

=SQL=
SELECT df.連番, df.G, df_1.連番, df_1.G
FROM df INNER JOIN df AS df_1 ON df.連番 +1= df_1.連番;

=pandas=
df2=pd.merge(df,df,how='left',left_on='連番'+1,right_on='連番')

発生している問題・エラーメッセージ

TypeError: can only concatenate str (not "int") to str

Answer 1

left_on の書き方を間違えています。
列名(文字列)に+1してもエラーです。列(Series)に+1したものを渡します。

df2 = pd.merge(
    df,
    df,
    how='left',
    left_on=(df['連番'] + 1),
    right_on='連番',
)

Comments

[tom57]

bsdfanさん
おっしゃる通りです。勉強し始めたばかりで基本が身についておらず
回答ありがとうございます。

Answer 2

効率性を不問にすればクロスジョインを使う方法もあるかと思います。

import pandas as pd
import numpy as np

df = pd.DataFrame({'連番': range(10), 'G': np.random.randint(0, 100, (10, ))})
dfx = pd.merge(df, df, how='cross').query('連番_x + 1 == 連番_y')\
        .set_axis(['df.連番', 'df.G', 'df_1.連番', 'df_1.G'], axis=1)

print(dfx)

df.連番	df.G	df_1.連番	df_1.G
0	46	1	14
1	14	2	18
2	18	3	29
3	29	4	92
4	92	5	73
5	73	6	3
6	3	7	57
7	57	8	33
8	33	9	15

Comments

[tom57]

melianさん、回答ありがとうございます。
確認しようとしたところ、crossでエラーが出てしまいました。
KeyError: 'cross'

[melian]

クロスジョインに対応しているのは Pandas version 1.2.0 からになります。おそらく、1.2.0 よりも前のバージョンをお使いなのではないでしょうか。

[tom57]

melianさん、ありがとうございます
おっしゃるとおりバージョン古かったです。アップデートします