Building up an array in numpy/scipy by iteration in Python?

后端 未结 3 860
我在风中等你
我在风中等你 2020-12-30 00:14

Often, I am building an array by iterating through some data, e.g.:

my_array = []
for n in range(1000):
  # do operation, get value 
  my_array.append(value)         


        
3条回答
  •  北海茫月
    2020-12-30 00:50

    NumPy provides a 'fromiter' method:

    def myfunc(n):
        for i in range(n):
            yield i**2
    
    
    np.fromiter(myfunc(5), dtype=int)
    

    which yields

    array([ 0,  1,  4,  9, 16])
    

提交回复
热议问题