Python DataFrames concat i append problem

Mam problem z datasets w pythonie. Próbuje przekopiować konkretnego wiersze do nowego dataframu ale nie mam pojęcia jak to zrobić.

mam dwie tablice:

pokemon_data

	#	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed
0	1	45	49	49	65	65	45
1	2	60	62	63	80	80	60
2	3	80	82	83	100	100	80
3	4	80	100	123	122	120	80
4	5	39	52	43	60	50	65
...	...	...	...	...	...	...	...
795	796	50	100	150	100	150	50
796	797	50	160	110	160	110	110
797	798	80	110	60	150	130	70
798	799	80	160	60	170	130	80
799	800	80	110	120	130	90	70

800 rows × 7 columns

combats_data


    First_pokemon	Second_pokemon	Winner
0	266	298	1
1	702	701	1
2	191	668	1
3	237	683	1
4	151	231	0
...	...	...	...
49995	707	126	0
49996	589	664	0
49997	303	368	1
49998	109	89	0
49999	9	73	0

50000 rows × 3 columns

Stworzyłem trzeci dataset:

output1

	HP0	Attack0	Defens0	Sp. Atk0	Sp. Def0	Speed0	HP1	Attack1	Defense1	Sp. Atk1	Sp. Def1	Speed1	Winner

Chce przekopiować dane z pokemon_data do output1 w porządku jak z combats_data.

HP0 i HP1 to HP pierwszego pokemona i HP drugiego pokemona.
Chce użyć tych danych do sieci neuronowych z TensorFlow do oszacowania prawdopodobieństwa wygrania przez jakiegoś pokemona.

Dzięki z góry za rady.

Chodzi Ci o takie coś?

import pandas as pd 


if __name__ == "__main__":
    pokemon_data = pd.DataFrame({
        "Id": [1, 2, 3, 4, 5],
        "HP": [45, 60, 80, 80, 39],
        "Attack": [49, 62, 82, 100, 52],
        "Defense": [49, 63, 83, 123, 43],
        "Sp. Atk": [65, 80, 100, 122, 60],
        "Sp. Def": [65, 80, 100, 120, 50],
        "Speed": [45, 60, 80, 80, 65]})
    combats_data = pd.DataFrame({
        "First_pokemon": [1, 2, 3], 
        "Second_pokemon": [2, 3, 4], 
        "Winner": [1, 0, 1]})

    output = pokemon_data.merge(combats_data, left_on="Id", right_on="First_pokemon")
    output = output.merge(pokemon_data, left_on="Second_pokemon", right_on="Id",
                          suffixes=("_pokemon1", "_pokemon2"))

    print(output)

Dokładnie tak. Wielkie dzięki.