pandas DataFrame按照列/行转为列表:df.values.tolist()

1、将某列转为列表

import pandas as pd

data = {'a': [1, 2, 3, 4],
        'b': [5, 6, 7, 8]}

df = pd.DataFrame(data)
   a  b
0  1  5
1  2  6
2  3  7
3  4  8

a_column_list = df['a'].values.tolist() # [1, 2, 3, 4]

2、按照行将df转为列表

import pandas as pd

data = {'a': [1, 2, 3, 4],
        'b': [5, 6, 7, 8]}

df = pd.DataFrame(data)
print(df)
   a  b
0  1  5
1  2  6
2  3  7
3  4  8


df_list = df.values.tolist()
print(df_list) # [[1, 5], [2, 6], [3, 7], [4, 8]]

发表评论

评论列表,共 0 条评论

    暂无评论