可以通過.columns和.index着兩個屬性返回數據集的列索引和行索引
設data是pandas的一個DataFram類型的數據集。
則data.index返回一個index類型的行索引列表,data.index.values返回的是行索引組成的ndarray類型。
則data.columns返回一個index類型的列索引列表,data.columns.values返回的是列索引組成的ndarray類型。
下面是代碼示例
1 import numpy as np 2 import matplotlib.pyplot as plt 3 import pandas as pd 4 from sklearn.preprocessing import StandardScaler 5 6 data = pd.read_csv("C:/學習/python/creditcard/creditcard.csv") 7 print(data.head()) 8 columns = data.columns 9 columns_value = columns.values 10 rows = data.index 11 rows_value = rows.values 12 print(type(columns), columns) 13 print(type(columns_value), columns_value) 14 print(type(rows), rows) 15 print(type(rows_value), rows_value)
運行結果如下
這里columns雖然是Index類型,但是columns != "Class"執行后,返回的是ndarray類型,元素是由“True"和"False"組成。所以可以根據這個語句找出數據集
中的指定行和列。