python中使用imshow和imsave查看和保存np三维数组数据

最新推荐文章于 2024-04-21 09:34:21 发布

原创最新推荐文章于 2024-04-21 09:34:21 发布 · 1.9k 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

文章介绍了如何利用matplotlib库中的imshow函数展示数据，包括对不同形状数组的支持，并展示了用法示例。同时，文章演示了使用imsave函数保存图像到指定文件，提供了保存单个数组切片的代码片段。

一、查看数据imshow

plt.imshow(
    X,
    cmap=None,
    norm=None,
    aspect=None,
    interpolation=None,
    alpha=None,
    vmin=None,
    vmax=None,
    origin=None,
    extent=None,
    shape=None,
    filternorm=1,
    filterrad=4.0,
    imlim=None,
    resample=None,
    url=None,
    *,
    data=None,
    **kwargs,
)

其中，X变量存储图像，可以是浮点型数组、unit8数组以及PIL图像，如果其为数组，则需满足一下形状：
(1) M*N 此时数组必须为浮点型，其中值为该坐标的灰度；
(2) M*N*3 RGB（浮点型或者unit8类型）
(3) M*N*4 RGBA（浮点型或者unit8类型）

import pandas as pd
import numpy as np
import matplotlib.pyplot as plot
labelSet = np.zeros((480,800,1597)) #返回一个给定形状和类型的用0填充的数组
for i in range(240):
    for j in range(400):
        for k in range(898):
            labelSet[2i][2j][2k]=1
for i in range(480):
    plot.imshow(labelSet[i, :, :].T, cmap='seismic', aspect='auto')

二、保存图片imsave

imsave（'filename'，array)

其中filename为保存的文件路径和名字，array为数组

for i in range(480):
    plot.imsave("data/image/testdata/horizon-test{}.png".format(i),labelSet[i, :, :].T)
    #plot.imshow(labelSet[i, :, :].T, cmap='seismic', aspect='auto')

标签

#python #开发语言 #numpy