首页 > 解决方案 > 尝试在 Python 中生成一个 128 x 128px 的正方形

问题描述

我正在从教程中做到这一点。这是代码:

import numpy as np
import matplotlib.pyplot as plt

size = (128,128)
image = np.zeros(size).astype(np.uint8)
image[1:64,1:64] = 255
image
np.array([[  0,   0,   0, ...,  0,  0,  0],
          [  0, 255, 255, ...,  0,  0,  0],
          [  0, 255, 255, ...,  0,  0,  0],
          ...,
          [  0,   0,   0, ...,  0,  0,  0],
          [  0,   0,   0, ...,  0,  0,  0],
          [  0,   0,   0, ...,  0,  0,  0]], dtype=np.uint8)

plt.imsave('foo.png', image, cmap="gray")

错误信息:

错误信息

谁能告诉我如何解决它?数组语法错了吗?

标签: pythonnumpy

解决方案


我认为您将部分输出复制到您的代码中。

这是正确的代码:

import numpy as np
import matplotlib.pyplot as plt

size = (128,128)
image = np.zeros(size).astype(np.uint8)
image[1:64,1:64] = 255

plt.imsave('foo.png', image, cmap="gray")

推荐阅读