numpy零如何實現參數形狀？

Question

我想實現一個類似的功能，並且想要接受一個數組或數字，我傳遞給numpy.ones。

具體來說，我想這樣做：

def halfs(shape): 
    shape = numpy.concatenate([2], shape) 
    return 0.5 * numpy.ones(shape) 

例輸入 - 輸出對：

# default 
In [5]: beta_jeffreys() 
Out[5]: array([-0.5, -0.5]) 

# scalar 
In [5]: beta_jeffreys(3) 
Out[3]: 
array([[-0.5, -0.5, -0.5], 
     [-0.5, -0.5, -0.5]]) 

# vector (1) 
In [3]: beta_jeffreys((3,)) 
Out[3]: 
array([[-0.5, -0.5, -0.5], 
     [-0.5, -0.5, -0.5]]) 

# vector (2) 
In [7]: beta_jeffreys((2,3)) 
Out[7]: 
array([[[-0.5, -0.5, -0.5], 
     [-0.5, -0.5, -0.5]], 

     [[-0.5, -0.5, -0.5], 
     [-0.5, -0.5, -0.5]]]) 

Answer 1

def halfs(shape=()): 
    if isinstance(shape, tuple): 
     return 0.5 * numpy.ones((2,) + shape) 
    else: 
     return 0.5 * numpy.ones((2, shape)) 



a = numpy.arange(5) 
# array([0, 1, 2, 3, 4]) 


halfs(a.shape) 
#array([[ 0.5, 0.5, 0.5, 0.5, 0.5], 
#  [ 0.5, 0.5, 0.5, 0.5, 0.5]]) 

halfs(3) 
#array([[ 0.5, 0.5, 0.5], 
#  [ 0.5, 0.5, 0.5]])