python - 遍历每列中的值

Question

如何遍历 DataFrame 中的数据，我想在移动到下一列之前遍历第一列中的所有数据。

使用下面的代码，我可以逐行遍历所有列的数据

相反，我想在移动到下一列之前遍历第一列中的所有数据。

import time
import pandas as pd


df = pd.read_csv('testDATA.csv')

for i, row in df.iterrows():
    #print(row)
    time.sleep(1)
    for j, column in row.iteritems():
        print(column)
        time.sleep(1)

print(df)

Answer 1

pandas.DataFrame 本身有 DataFrame.iteritems()迭代数据框所有列的方法:

In [8]: df = pd.DataFrame({'col1': [1, 2], 'col2': [3, 4]})                                                 

In [9]: for col_name, col_data in df.iteritems(): 
   ...:     print(col_name) 
   ...:     print(col_data.values)   # can be iterated separately 
   ...:                                                                                                     
col1
[1 2]
col2
[3 4]