我想從csv文件創建矩陣

Question

我想從CSV文件創建一個矩陣。

這是我嘗試過的：

df = pd.read_csv('csv-path', usecols=[0,1], names=['A', 'B'])
pd.pivot_table(df,columns='A', values='B')

輸出： [9197337 rows x 2 columns] 。

我想減少行數，就像我要創建前100個條目或1000個條目的矩陣一樣。我該怎么做？

Answer 1

由於csv模塊僅處理完整的文件，因此在使用之前提取感興趣的行是最容易的。 您可以在使用Unix head實用程序運行程序之前執行此操作。 這是應該在Python中工作的一種方式：

with open("csv-path") as inf, open("mod_csv_path", "w") as outf:
    for i in range(1000):
        outf.write(inf.readline())

顯然，您將讀取“ mod_csv_path”而不是“ csv-path”作為輸入文件。

Answer 2

熊貓似乎是正確的做法？ 您能否提供CSV文件的示例。

另外，對於熊貓，您可以限制數據框的大小：

limited_df = df.head(num_elements)