Python循环中如何提取指定范围图像并保存为.npy文件?
提取指定范围图像并保存为.npy文件
不用通过循环逐个处理,直接利用numpy数组的切片功能就能高效完成需求,步骤如下:
提取目标范围的图像数组
由于Python切片遵循左闭右开规则,x1[25:32]会直接提取索引从25到31的所有图像(共8张),得到维度恰好为(8,128,128)的数组。保存为.npy文件
使用numpy的save()函数将提取后的数组以二进制格式保存为.npy文件。
完整代码示例:
import numpy as np # 提取25-31索引的图像(对应需求的25-32范围) extracted_imgs = x1[25:32] # 可选:验证数组维度是否符合预期 print(extracted_imgs.shape) # 输出应为 (8, 128, 128) # 保存到指定文件 np.save('target_images.npy', extracted_imgs)
后续若需加载该文件,执行以下代码即可:
loaded_imgs = np.load('target_images.npy')
内容的提问来源于stack exchange,提问作者Bodasheera
相关产品推荐
相关产品推荐

