如何使用 python 保存 to_csv

Question

我想將這些項目保存到 to_csv 作為一個文檔

df = pd.DataFrame({'STREAM':['EAGLE','HAWK','HAWK','HAWK','EAGLE','HAWK','EAGLE'],'MAT':['A','D','F','D','C','C','E'],'KIS':['B','D','E','D','A','C','D'],'GEO':['B','C','E','E','F','A','B'],'BST':['C','E','D','D','B','F','C']})

columns = ["A",'A-',"B","C","D","E", "F"]
a = df.melt(id_vars=['STREAM'], value_vars=['MAT','KIS','BST','GEO']).pivot_table(index='STREAM', columns='value', values='variable', 
 aggfunc='count', fill_value=0, margins=True, margins_name='TOT').rename_axis(None)

print('SHOW FIRST TEXT HERE')
print(a)
print()
print('SHOW SECOND TEXT HERE')
print(df)

這樣我的結果會是這樣的

       A  B  C  D  E  F  TOT
EAGLE  2  4  3  1  1  1   12
HAWK   1  0  3  6  4  2   16
TOT    3  4  6  7  5  3   28

SHOW SECOND TEXT HERE
STREAM MAT KIS GEO BST
EAGLE   A   B   B   C
HAWK    D   D   C   E
HAWK    F   E   E   D
HAWK    D   D   E   D
EAGLE   C   A   F   B
HAWK    C   C   A   F
EAGLE   E   D   B   C

任何有線索的人都可以提供幫助

Answer 1

您可以連接兩個數據框：

pd.concat([a, df], axis=0).to_csv('concatenated_dataframes.csv')

雖然不同的形狀意味着你有 NaN 填補“空白”

Answer 2

保存到 csv 通常意味着保存一個具有相同標題的表。 如果您想保持數據框的形狀， 保存到 MS Excel是一個不錯的選擇，因為您可以將它們保存在不同的選項卡或電子表格中。

如果你喜歡，你可以使用pickle 。 這將允許您將每個數據幀保存到泡菜中，也許在字典中。

import pickle
dict = ('first': df, 'second': a)

pickle.dump(dict, open( "yourfile.pickle", "wb" ))

然后打開備份。 pickle.load 打開文件。 在這種情況下，您將兩個數據幀存儲在一個字典中，因此當您從 pickle 打開/加載文件時，您將獲得一個包含兩個數據幀的字典。

new_dict = pickle.load( open( "yourfile.pickle", "rb" ) )

# To then get your dataframes, 
df = new_dict['first'] 
a = new_dict['second']

HDF5 可能比您需要的更復雜。