使用Pandas
安裝
pip install pandas
DataFrame
一個(gè)DataFrame可以看成一個(gè)二維表格，就像excel一樣浓利。
import numpy as np
import pandas as pd
data = np.array([['', 'Col1', 'Col2'],
['Row1', 1, 2],
['Row2', 3, 4]])
df = pd.DataFrame(data=data[1:, 1:],
index=data[1:, 0],
columns=data[0, 1:])
print(df)
輸出
Col1 Col2
Row1 1 2
Row2 3 4
獲取DataFrame中的值
print(df.iloc[1][1])
# 4
print(df.loc['Row2']['Col2'])
# 4
print(df.iloc[1,])
# Col1 3
# Col2 4
# Name: Row2, dtype: object
print(df.iloc[:,1])
# Row1 2
# Row2 4
# Name: Col2, dtype: object
從csv讀取
data.csv
,2014,2015,2016
營(yíng)業(yè)收入,1.25,1.87,2.2
營(yíng)業(yè)成本,0.4,0.5,0.6
凈利潤(rùn),0.82,1.3,1.7
?df = pd.read_csv('data.csv', index_col=0)
print(df)
輸出
2014 2015 2016
營(yíng)業(yè)收入 1.25 1.87 2.2
營(yíng)業(yè)成本 0.40 0.50 0.6
凈利潤(rùn) 0.82 1.30 1.7
