python - 使用 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])