从数组的numpy数组中删除零值

Question

原始数据

 [[ 0.00000000e+00  1.00000000e+00 -6.76207728e+00 -1.63236398e+01]
 [ 0.00000000e+00  1.00000000e+00  2.51283367e+01  1.13952157e+02]
 [ 0.00000000e+00  1.00000000e+00  3.11402956e+00 -5.16009612e+02]
 [ 0.00000000e+00  1.00000000e+00  3.10969787e+01  1.82175649e+02]
 [ 1.00000000e+00 -2.31269114e+00 -4.13720127e+02  3.55395844e+03]
 [ 1.00000000e+00  4.54598490e+01  6.19694322e+02  2.61091335e+03]
 [ 1.00000000e+00  7.36925014e-01 -4.49386738e+02 -1.22392549e+03]
 [ 1.00000000e+00  3.29511609e+00 -4.43413555e+02 -4.12677155e+03]]

我试着用下面这首颂歌去掉零

def removeZeroPadding(X):
  res = []
  for poly in enumerate(X):
      tmp = poly[1]
      tmp = tmp[tmp != 0]
      res.append(tmp)
  return res

这转化为

[array([  1.        ,  -6.76207728, -16.32363975]), array([  1.        ,  25.1283367 , 113.95215706]), array([   1.        ,    3.11402956, -516.0096117 ]), array([  1.        ,  31.09697873, 182.17564943]), array([ 1.00000000e+00, -2.31269114e+00, -4.13720127e+02,  3.55395844e+03]), array([1.00000000e+00, 4.54598490e+01, 6.19694322e+02, 2.61091335e+03]), array([ 1.00000000e+00,  7.36925014e-01, -4.49386738e+02, -1.22392549e+03]), array([ 1.00000000e+00,  3.29511609e+00, -4.43413555e+02, -4.12677155e+03])]

如何保持原始数据的结构不带零？谢谢

编辑：应该是这样的

 [[ 1.00000000e+00 -6.76207728e+00 -1.63236398e+01]
 [ 1.00000000e+00  2.51283367e+01  1.13952157e+02]
 [ 1.00000000e+00  3.11402956e+00 -5.16009612e+02]
 [ 1.00000000e+00  3.10969787e+01  1.82175649e+02]
 [ 1.00000000e+00 -2.31269114e+00 -4.13720127e+02  3.55395844e+03]
 [ 1.00000000e+00  4.54598490e+01  6.19694322e+02  2.61091335e+03]
 [ 1.00000000e+00  7.36925014e-01 -4.49386738e+02 -1.22392549e+03]
 [ 1.00000000e+00  3.29511609e+00 -4.43413555e+02 -4.12677155e+03]]

Answer 1

看看https://numpy.org/doc/stable/reference/generated/numpy.trim_zeros.htmltrim_zeros

它将避免for循环

Answer 2

似乎您不想要数组列表作为输出，而是想要一个二维数组（即[[...]]而不是[array([...]), array([...])]）

但是，这是不可能的，因为通过删除零来修剪数组中的行后，这些行的大小会有所不同（，即有些行最终有3个元素，有些行最终有4个元素）。如果希望它是单个数组，就像数学中的矩阵一样，所有列（和所有行）都需要具有相同数量的元素

或者，您可以为要修剪的元素指定不同的值，例如None，或者您可以create an empty array and fill it

Answer 3

你所追求的根本是不可能的。您不能同时从每一行中删除数量可变的条目，同时保证每一行具有相同数量的条目（这是获取类型[[...]]的ndarray所必需的）。hlzl's answer对此进行了更详细的解释

以下函数将从每行的左侧和右侧删除0-项，但返回的类型为[array([...]), array([...])]：

def trim_zeros_in_rows(ndarr):
    return np.array([np.trim_zeros(row) for row in ndarr])

print(trim_zeros_in_rows(np.array([[0., 0., 1., 0.],
                                   [2., 3., 4., 0.]])))

印刷品：

[array([1.]) array([2., 3., 4.])]