for-loop - Numpy savetxt 循环

Question

使用 Numpy，我将拆分一个维度数组 (557124,2) , dtype = "S10" , 在 6 个子数组中使用:

sub_arr = np.split(arr, 6)

现在我想使用 for循环 savetxt并将 6 个子数组保存到 6 个 .txt 文件中。

我试过:

for i in sub_array:
    np.savetxt(("Subarray", i, ".txt"), sub_array[i], fmt='%s')

有2个问题:

说 in sub_array 是不对的.我应该使用 range(5)但我想让它适应任意数量的子数组。

当我做 ("Subarray", i, ".txt") 时，我想我可以像在 R 中一样使用一种“粘贴”。 . Numpy 中有类似的东西吗？

有什么想法吗？

Answer 1

据我了解

sub_arr = np.split(arr, 6)

返回 6 个 numpy 数组的列表。然后你可以使用 enumerate 来获取每个数组及其索引

fname_template = "Subarray.{i}.txt"
for i, sarr in enumerate(sub_arr):
    np.savetxt(fname_template.format(i=i), sarr, fmt='%s')

要创建文件名，我使用了 new string formatting .否则你可以用 + 连接字符串如 "Subarray."+str(i)+".txt" ，但您必须确保您连接的所有元素都是字符串。