python - Numpy unicode 转 int

Question

通过 Pycharm 使用 python 2.7

我有一个 python 代码 - 非常简单。

a = [2, 3, 4, 5]

b = [30,32,33,34]
c = ['test1', 'test2', 'test3', 'test4']
a = np.asarray(a) # -- Similarly for b and c

我将这些列表转换为 numpy 数组。

然后我做了以下事情

d = np.transpose(np.vstack((a,b,c)))

通过它我得到了一个 3D numpy 数组。

当我打印 d 时，我在每个元素之前看到一个 u"，这可能表示 unicode 数据类型。

现在我如何访问 a,b 的列并对这些值进行一些算术运算。每当我尝试访问一个值并将其转换为 int 时，都会显示 以 10 为基数的 int() 的文字无效。

我所需要的只是读取第一列或第二列——进行一些算术运算。第一列和第二列值为 a 和 b

Answer 1

因为最后一列是文本，所以其他两列也转换为文本。一个矩阵只能有一种类型。解决方案是结构化数组:

a = [2, 3, 4, 5]
b = [30,32,33,34]
c = ['test1', 'test2', 'test3', 'test4']
matrix = np.array(zip(a,b,c), dtype=[('a',int),('b',int),('c','S5')])

要访问这些值，您必须使用列名称:

matrix['a'][2:4]

或

matrix[5]['b']