我想从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)