numpy.array转换为图片并显示出来

  统计/机器学习 计算机视觉 数据可视化 Python 数据竞赛    浏览次数:23329        分享
0

我想知道在python里有什么方法可以让numpy array(数值矩阵)转为图片并且显示?

就好比这个数据竞赛里的两个图片

就是根据两个40乘以40的numpy array得到的图片


 

剪叔   2017-12-29 14:57



   2个回答 
7


# ipython@python3环境
import pandas as pd
import numpy as np

with open('~/train.csv') as f:
    data = pd.read_csv(f)

X = data.drop(['y','id'], axis=1)
X = np.array(X)

#  每一行代表一张图片,选取第一行。
pic1 = X[1]
# 图片分辨率是40*40
pic1 = pic1.reshape((40, 40))

# 绘图
%matplotlib inline
from matplotlib import pyplot as plt
# 传入cmap='gray'指定图片为黑白
plt.imshow(pic1, cmap='gray')

另外补充一下,如果是彩色图片,在reshape的时候需要设置成三维,分别代表每个像素点RGB三种配色的取值:


pic1.reshape((height, width, 3))


SofaSofa数据科学社区DS面试题库 DS面经

s3040608090   2017-12-29 15:43

您的回答非常具体!谢谢! - 剪叔   2017-12-29 23:48
2


import matplotlib.pyplot as plt
plt.imshow(my_figure, cmap=plt.cm.binary)
plt.show()


SofaSofa数据科学社区DS面试题库 DS面经

数据痴汉   2018-03-23 01:19



  相关讨论

大数据建模、分析、挖掘技术应用研修班

有大佬可以分享一下竞赛的TOP解决方案吗

请教一下:关于融合stack

时下最流行的模型

模型融合问题

大数据建模、分析、挖掘技术应用研修

随机森林中的predict_proba

如何在本地判断算法效果得到优化?

怎么把3D的numpy array显示为彩色图像并且保存?

用投票融合(VotingClassifier)的过程中出现问题,求帮助?

  随便看看

pandas报错ValueError: Cannot convert non-finite values (NA or inf) to integer

单一变量下的异常检测该怎么做?

opencv里waitkey和destroyAllWindows有什么用?

激活函数RELU在0点的导数是多少?

如何调节seaborn里字体大小?