在Python中将csv文件导入矩阵/数组

Question

我试图将包含字符串和数据数字矩阵的大型csv文件导入到Python中的数组/矩阵中.在MATLAB中,我曾经加载文件并简单地将其分配给矩阵,但在Python中似乎有点棘手.有人可以提出建议吗？我是Python的新手.谢谢大家.

Answer 1

您可以使用内置的csv模块将数据加载到多维列表:

import csv

with open('data.csv', 'rb') as f:
    reader = csv.reader(f)
    data_as_list = list(reader)

print data_as_list
# [['data1', 1],
#  ['data2', 2],
#  ['data3', 3]]

Answer 2

您可以使用熊猫。

import pandas as pd
df = pd.from_csv('filename.csv')

如果定界符不是“”，则可以使用sep关键字更改默认值，例如：

df = pd.from_csv('filename.csv', sep='\')

您将获得一个带有强大分析功能的数据框。