我以前用的都是 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]])