python - 在 Python 的每次迭代中更改名称以保存

Question

我想知道如何在 for 循环中更改文件的名称，以便用不同的连续名称保存它。例如

for i in range(0, np.max(idx)+1):    
    vars()['cluster' + str(i)]= [tracks[x] for x in range(len(tracks)) if idx[x] == i]
    function_to_write("/Users/George/Desktop/file.bmp", vars()['cluster' + str(i)])  

如何将文件名更改为 file1、file2 等？

Answer 1

你可以这样做:

"/Users/George/Desktop/file{0}.bmp".format(i+1)

这使用 python 的 string.format(...) 方法将 {} 替换为指定值。

{0} 表示替换它的值将是 format(...) 中的第 0 个参数(第 1 个位置)

此外，我看到您正在使用 vars()，这意味着您可能将变量存储为 cluster1、cluster2、cluster3 ，...

这通常不是一个好主意，相反，建议您使用 list。

如果您将所有“集群”存储在一个名为 clusters 的列表中，那么您可以改为:

for i in range(0, np.max(idx)+1):
    clusters[i] = [tracks[x] for x in range(len(tracks)) if idx[x] == i]
    function_to_write("/Users/George/Desktop/file{0}.bmp".format(i+1), clusters[i])