nparray的维度和取值方式

「已注销」

于 2019-08-15 10:35:18 发布

阅读量2.3k

点赞数

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_43014927/article/details/99623582

本文通过示例代码深入探讨了Python中numpy库的nparray数据结构，重点关注其维度处理和元素访问方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

运行下面的代码，分析结果，不做解释。

import numpy as np
 
def func(img, label):
	#print('label[0]:', label[:,:,0])
	if (1):
		img = img / 255.
		label = label[:, :, :, 0] if (len(label.shape) == 4) else label[:, :, 0]
		new_label = np.zeros(label.shape + (3,))
		for i in range(3):
			new_label[label == i, i] = 1
			print('\n i = \n ', i, '\n label = \n', label, '\n label==i: \n',label==i)
			print('\n\n after-img:\n',img,'\n\n after-label:\n', new_label, '\n')
		label = new_label
 
	elif (np.max(img) > 1):
		img = img / 255.
		label = label / 255.
		label[label > 0.5] = 1
		label[label <= 0.5] = 0
	return (img, label)
	
label = np.array([[[1,2,0],
				   [0,1,2],
				   [2,1,0]],
				   
				  [[1,2,0],
				   [2,1,2],
				   [2,1,0]],
				   
				  [[0,2,0],
				   [0,1,2],
				   [2,1,0]]])
				   
img = np.array([[129, 255, 30],
				[30, 30, 99],
				[90, 123, 49]])
 
#print(len(label.shape))
print('\n\n before-img:\n',img,'\n\n before-label:\n', label, '\n')
func(img, label)