如何在Python中切片数组

时间:2017-09-27 16:58:35

标签: python arrays python-2.7 list array-splice

我正在尝试在python上编写自己的机器学习脚本(我知道有这样的库,但这纯粹是为了它的乐趣 - 我正在学习Python)。我有以下数组;

[array([[[  5,   5,   5, 255],
    [  6,   6,   6, 255],
    [  6,   6,   6, 255],
    ..., 
    [ 12,  12,  12, 255],
    [ 10,  10,  10, 255],
    [ 10,  10,  10, 255]],

   [[  8,   8,   8, 255],
    [ 10,  10,  10, 255],
    [ 14,  14,  14, 255],
    ..., 
    [ 15,  15,  15, 255],
    [ 13,  13,  13, 255],
    [ 13,  13,  13, 255]],

这种情况持续了一段时间。我使用以下代码获得了这个数组:

imagesList = listdir("someaddress")
loadedImages = []
for image in imagesList:
    #img = PImage.open()
    loadedImages.append(misc.imread("someaddress" + image))

我的逻辑是,我想在图像文件中读取像素值数组,以用于图像分类问题。从上面的数据可以看出,图像是灰度级的。我想从这些数据中删除一个维度,每个像素只有一个值(例如,([[[5],[6],[6],[12] ......)255只是alpha值(我不关心)。我知道这是我需要使用的数组拼接,但是男孩我不知道如何将它应用于这个问题。

我试过了; loadedImages [:,1]

我收到以下错误;

TypeError: list indices must be integers, not tuple

我真正想要的结果如下:

[array([[  5,
  6,
  6,
..., 
 12,
 10,
 10],

 [  8,
   10,
   14,
    ..., 
   15,
   13,
   13,

2 个答案:

答案 0 :(得分:2)

只需使用numpy就可以进行你想要做的扩展切片:

import numpy as np
a = np.array([[  5,   5,   5, 255],
 [  6,   6,   6, 255],
 [  6,   6,   6, 255]])
# first column:
print(a[:, 0])

在此示例中,:表示数组的所有行,0表示第一列

答案 1 :(得分:1)

您是否尝试过删除拼接上的逗号,例如loadedimages [:1]?对于此数组arr = [ 5, 5, 5, 255],要删除255,您可以执行arr = arr[:3]。对于多维数组,您可以根据需要嵌套for循环,以便迭代各个数组。 e.g。

main_array = [[  5,   5,   5, 255],
              [  6,   6,   6, 255],
              [  6,   6,   6, 255]]

for sub_array in main_array:
  spliced_array = sub_array[:3]
  print spliced_array

将产生

[5, 5, 5]
[6, 6, 6]
[6, 6, 6]

此外,在Python中,这种数据结构称为列表。这可能是您很难找到相关信息的原因。

编辑:您也可以尝试使用列表推导。它在python中非常有用。例如

arr = [[  5,   5,   5, 255],
       [  6,   6,   6, 255],
       [  6,   6,   6, 255]]

filtered_arr = [a[0] for a in arr]
filtered_arr
>>>[5,6,6]