请问python中,如何抓取dataframe的列数和第n列的数据?

我以前用的都是 data['AttributeName']抓取某列的数据,想求个抓取列数第n列数据方法。

df.shape[1] 可以得到列数
df.iloc[:, [n]] 第n列

就是你这个用法,你只需得到它的列数 就行了
import numpy as np
import pandas as pd

d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)
columns = df.columns.values
print(columns)
print("*"*50)
for i in columns:
# print(df[i])

print("="*50) #或者直接想得到第几列的数据 就根据columns的下标位置取
for i in range(0,len(columns)):
# print(df[columns[i]])