读取csv文件
with open('results.csv') as f:
data=pd.read_json(f)
读取json文件:
with open('results.json') as f:
data=pd.read_json(f)
此时data是dataframe类型:pandas.core.frame.DataFrame
可以通过.iloc 和 .loc 选取数据,如:
- data_test1=data.iloc[:,:8] #选取位置为[0,8)列的整列数据
- data_test2=data.iloc[0:2,8] #选取位置为8的列的[0,2)行的数据
- data_test3=data.loc[0:2,'工龄'] #选取列名为‘工龄’的[0,2]行的数据
也可以把data(dataframe类型)直接转换成矩阵:
data=np.array(data)