python使用两列合并两张表格_Python如何将数据框的两列合并为一个列表？

I have a dataframe as given below

df =

index data1 data2

0 20 120

1 30 456

2 40 34

How to combine two columns in above df into a single list such that first row elements come first and then second row.

My expected output

my_list = [20,120,30,456,40,34]

My code:

list1 = df['data1'].tolist()

list2 = df['data2'].tolist()

my_list = list1+list2

This did not work?

解决方案

That doesn't work since it won't add by same index, use the below list comprehension:

print([x for i in zip(df['data1'], df['data2']) for x in i])