Python 使用numpy.genfromtxt读取单行CSV

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

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

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

这适用于包含多行的文件。但无法处理此文件：

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）。有人知道如何修复它吗？

强制将其放入二维数组：

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

可能重复的