Pandas數據的保存與加載

原文地址

分類目錄——Pandas

• 導入支持包

  import pandas as pd #加載模塊
  

• 生成測試數據

  df = pd.DataFrame({'col1':[11,12,13], 'col2':[21,22,23]})
  

• 保存到.csv文件

  df.to_csv('data/savecsv.csv')
  

• 從.csv文件中讀取

  # 從.csv文件讀取讀取DataFrame對象
  df_csv = pd.read_csv(r'data/savecsv.csv', sep=',', header=0, index_col=0)
  # 這裏的sep是分隔符，當爲‘,’時，可以在excel中方便的查看；爲‘\t’時，反而在excel不能很好的顯示了
  # header    # 索引行，指定那一行是索引， 默認值0
  # index_cols    # 索引列，指定那一列是做引， 默認值None， 即沒有索引列， =-1代表最後一列
  print(type(df_csv))
  # <class 'pandas.core.frame.DataFrame'>
  print(df_csv)
  #    col1  col2
  # 0    11    21
  # 1    12    22
  # 2    13    23
  

• 將DataFrame對象轉換成numpy矩陣

  # 將DataFrame對象轉成np.array對象
  npdata2 = df_csv.values
  print(type(npdata2))
  # <class 'numpy.ndarray'>
  print(npdata2)
  # [[11 21]
  #  [12 22]
  #  [13 23]]
  

• DataFrame對象數據保存到excel並讀取

  df.to_excel('data/saveexcel.xls')
  df_excel = pd.read_excel('data/saveexcel.xls', header=0, index_col=0)
  # sheet_name=0  # 指定sheet
  # header    # 索引行，指定那一行是索引， 默認值0
  # index_cols    # 索引列，指定那一列是做引， 默認值None， 即沒有索引列， =-1代表最後一列
  print(df_excel)
  #    col1  col2
  # 0    11    21
  # 1    12    22
  # 2    13    23
  

• 參考文獻

  Pandas 導入導出