Python如何将数据框的两列合并为一个列表? [英] Python how to combine two columns of a dataframe into a single list?

问题描述

我有一个如下所示的数据框

I have a dataframe as given below

df = 

index    data1    data2
0         20       120
1         30       456
2         40       34

如何将df上方的两列合并为一个列表，以使第一行元素排在第一位，然后是第二行.

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

我的预期输出

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

我的代码:

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

my_list = list1+list2

这行不通吗?

推荐答案

这不会起作用，因为它不会按相同的索引添加，请使用以下列表理解:

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])

这篇关于Python如何将数据框的两列合并为一个列表?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！