python - Numpy 数组中的多个数据类型

Question

我在一个 numpy 数组中有以下数据集:

数组 1:

[[a, 1, 20]
 [a, 3, 40]
 [b, 1, 20]
 [b, 2, 40]
 [c, 5, 90]]

数组 2:

[[a, 2]
 [a, 5]]

我要完成的是以下内容:Array2[0,0]=a 和 Array2[0,1]=2我想从第一个数组进行插值以找到 a,2,30。

为此，我使用了 np.where(Array1==item)[0] 来查找 'a'，但我无法进行插值，因为用于导入的 dtype 是一个字符串，而不是一个 int。

自从我使用 Numpy 以来已经有一段时间了，所以如果我完全陷入困境，请告诉我。

Answer 1

我不完全清楚你想做什么，但听起来你想指定一个聚合数据类型。

这在 dtype 中有详细解释。文档。

例如，这里有一种方法可以指定每行有一个 1 个字符的字符串和一个 64 位 native float (当您不关心字段名称是什么时):

dt = np.dtype('U1, f8')

当然还有其他的写法；阅读整页了解详情。

并且，假设您已经使用 loadtxt 阅读了这篇文章，那里的文档有一个使用这种数据类型的很好的例子。例如:

>>> s2 = 'a 2\na 5\n'
>>> i2 = io.StringIO(s2)
>>> a2 = np.loadtxt(i2, 'U1, i4')
>>> a2
array([('a', 2), ('a', 5)],
      dtype=[('f0', '<U1'), ('f1', '<i4')])