对 CSV 应用一个简单的函数并保存多个 csv 文件

Question

我试图通过将每个值与一系列值相乘来复制数据，并将结果保存为 CSV。

我创建了一个函数“Replicate_Data”，它接受输入的 numpy 数组并乘以一个范围之间的随机值。创建 100 个文件并保存为 P3D1 、 P4D1 等的最佳方法是什么。

def Replicate_Data(data: np.ndarray) -> np.ndarray:
    Rep_factor= random.uniform(-3,7)
    data1 = data * Rep_factor
    return data1

P2D1 = Replicate_Data(P1D1)
np.savetxt("P2D1.csv", P2D1, delimiter="," , dtype = complex)

Answer 1

这是一个您可以用作参考的示例。

我生成名为toy 的玩具数据，然后我使用np.random.uniform 生成n 随机值并将其命名为randos，然后我将这两个对象相乘以使用numpy 广播形成out。您也可以在循环中执行此乘法（实际上与您保存的相同）：根据输入数组的大小，它可能会占用大量内存，因为我已经编写了它。更完整的答案可能取决于您输入数据的形状。

import numpy as np
toy = np.random.random(size=(2,2)) # a toy input array
n = 100 # number of random values
randos = np.random.uniform(-3,7,size=n) # generate 100 uniform randoms
# now multiply all elements in toy by the randoms in randos
out = toy[None,...]*randos[...,None,None] # this depends on the shape.
# this will work only if toy has two dimensions. Otherwise requires modification
# it will take a lot of memory... 100*toy.nbytes worth

# now save in the loop.. 
for i,o in enumerate(out):
    name = 'P{}D1'.format(str(i+1))
    np.savetxt(name,o,delimiter=",")


# a second way without the broadcasting (slow, better on memory)
# more like 2*toy.nbytes
#for i,r in enumerate(randos):
#    name = 'P{}D1'.format(str(i+1))
#    np.savetxt(name,r*toy,delimiter=",")