代码之家  ›  专栏  ›  技术社区  ›  ddccffvv

如何在numpy中创建多维矩阵

  •  -1
  • ddccffvv  · 技术社区  · 8 年前

    我试图用numpy创建一个多维矩阵来描述一个RGB图像。像这样的事情是行不通的:

    完全数字(100,100,[0,0,0])

    这失败了 TypeError: data type not understood . 我想得到的是一个像素矩阵,每个像素都有RGB值。

    编辑:这让我走到了一半:

    n = numpy.empty((3,3,3))
    n[:] = [0,0,0]
    

    然而,这为每个点提供了一个浮点数组,而uint8数组就足够了。我该怎么解决?

    1 回复  |  直到 8 年前
        1
  •  0
  •   ddccffvv    8 年前

    很明显,这就是诀窍:

    n = numpy.empty((3,3,3))
    n[:] = numpy.array([0,0,0], dtype=numpy.uint8)