使用 numpy.genfromtxt 读取单行 CSV

Question

我正在使用以下脚本使用 numpy 从标准输入读取文件。

#!/usr/bin/env python
import numpy as np
import sys

data = np.genfromtxt(sys.stdin, delimiter=",")
print data.shape
print data

这适用于超过 1 行的文件。但无法为该文件工作：

1,2,2,2,2,2,1,1,1

我是这样运行的

$ cat input-file.txt | ./test.py

输出如下：

(9,)
[ 1.  2.  2.  2.  2.  2.  1.  1.  1.]

它应该具有形状 (,9)。有人知道怎么解决吗？

Answer 1

强制转换成二维数组：

data = np.genfromtxt(sys.stdin, delimiter=",")
if len(data.shape) == 1:
    data = np.array([data])