Python循环构造列表时numpy.piecewise返回array转普通数值问题

你遇到的问题是np.piecewise即使传入单个数值，返回结果也是0维numpy数组，而非Python原生数值，所以列表输出会显示array(1)这类格式。以下是两种可行的解决方法：

方法1：修改自定义函数的返回值（推荐） #

在函数返回时调用numpy数组的.item()方法，直接将0维numpy数组转为Python原生数值，修改后的代码如下：

import numpy as np
def f(num):
  return np.piecewise(num, [(num+1)%3==1, (num+2)%3==0,(num+1)%3==0], [1,-1,-1]).item()
print(f(0))
print(f(1))
fx = [[f(i),i,(i+1)%2] for i in range(2)] 
print(fx)

如果需要固定输出浮点型数值，把.item()替换为float()即可，最终输出就是你需要的：

1
-1
[[1.0, 0, 1], [-1.0, 1, 0]]

方法2：构造列表时做转换 #

如果你不想修改原函数f的逻辑，可以在列表推导式里对f(i)的返回结果单独做类型转换：

# 转原生数值用.item()
fx = [[f(i).item(), i, (i+1)%2] for i in range(2)]
# 转浮点型用float()
fx = [[float(f(i)), i, (i+1)%2] for i in range(2)]

内容的提问来源于stack exchange，提问作者Schrodinger