Question

我想读取csv文件中的数据并执行某种格式。但是我将数据重新格式化为numpy数组后，忘记了如何删除“ ''”。我已经看到有一种技巧可以在一行之前使用eval()。但是我现在找不到它。

我的csv文件数据如下：

6,1 2 3 4 5 6 ... 2300 2301 2302 2303 2304 \ n

...

1,1 2 3 4 5 6 ... 2300 2301 2302 2303 2304 \ n

from load_data import load
import numpy as np
import matplotlib.pyplot as plt

with open('fer2013.csv') as f:
    f.readline()
    line = f.readline()
    line = line.replace("\n","")
    line = line.split(",")
    X = np.array(line[1].split(" "))
    X.reshape((48,48))
    print(X)

输出为：

[['70' '80' '82' ... '52' '43' '41'] ['65' '61' '58' ... '56' '52' '44'] ['50' '43' '54' ... '49' '56' '47'] ... ['91' '65' '42' ... '72' '56' '43'] ['77' '82' '79' ... '105' '70' '46'] ['77' '72' '84' ... '106' '109' '82']]

但是我想要的是：

[[70 80 82 ... 52 43 41] ... [77 72 84 ... 106 109 82]]

我已经看到了一些类似这种格式的技巧（但实际上并非如此）：

X = np.array(for I in eval(line[1].split(" ")))

Answer 1

使用X = np.array(line[1].split(" ")).astype(np.int)

Answer 2

好，我找到了答案。

X = np.array(list(map(eval,line[1].split(" "))))

如何在不带引号的情况下将CSV文件读取为numpy数组

2 个答案: