删掉nd array数组中的所有零元素

  • 1、将数组转化为列表形式。
  • 2、找出列表中所有0元素的索引。
  • 3、使用 del 删掉列表中的0元素。
  • 4、将列表中的元素转化为整型。(列表中的元素为浮点型)
  • 5、将最终得到的不含0元素的列表转化成数组。
a = np.array([1, 2, 3, 4, 0, 5, 6, 0, 2, 0])
# 1、将数组转化为列表形式。
a_list = list(a)
# 2、找出列表中所有0元素的索引。
index = []
for j, each in enumerate(a_list):
    if each == 0:
        index.append(j)
        # 3、使用 del 删掉列表中的0元素。
        del a_list[j]
# 4、将列表中的元素转化为整型。
for k, each in enumerate(a_list):
    each_int = int(each)
    a_list[k] = each_int
# 5、将最终得到的不含0元素的列表转化成数组。
a = np.array(a_list)

你可能感兴趣的:(python,列表,numpy,索引,python)