我可以通过举一个例子来最好地提出我的问题。假设我想使用列表推导从两个循环生成一组 3 元素元组，如下所示：
[ (y+z,y,z) for y in range(10) if y%2==0 for z in range(20) if z%3==0 ]
这有效，给我
[(0, 0, 0), (3, 0, 3), (6, 0, 6), (9, 0, 9), (12, 0, 12), (15, 0, 15), ... ]
不过，我想知道是否有一种方法可以更干净地做到这一点，例如
[ (x,y,z) for y in range(10) if y%2==0 for z in range(20) if z%3==0 ... somehow defining x(y,z) ... ]
我认为像这样的东西更干净，特别是因为我真正需要做的事情比我在这里给出的例子要复杂得多。我尝试过的所有方法都给出了语法错误。
我有一个主程序和一个模块。我希望主程序在正常情况下使用模块的某些部分，但如果使用“DEBUG”cpp 宏编译，则使用一些添加的元素。这是我发现的使这项工作有效的方法：
program myprogram
use comp_module, only: data1, routine1 &
#idef DEBUG
, debug_data, debug routine
#endif
;
....
如果我不在末尾添加分号，编译器会在未编译进行调试时认为这是一个错误。所以，这有效。
只是它看起来有点不优雅，调试行中有一个前导逗号，后面有一个无用的分号。我可以看到一些开发人员稍后出现，不明白发生了什么，并重新引入编译器错误。
有什么更好的办法呢？
psi我有一个形状为 (3,1000) 的numpy 数组
psi.__class__
Out[115]: numpy.ndarray
psi.shape
Out[116]: (3, 1000)
我想部分填充psi另一个数组b
b.__class__
Out[113]: numpy.ndarray
b.shape
Out[114]: (3, 500)
我可以用循环来做到这一点：
for n in range(3):
psi[n][:500] = b[n]
但在我看来，应该有一种更直接的方法来做到这一点。但例如
psi[:][:500] = b
因错误而失败
Traceback (most recent call last):
File "<ipython-input-120-6b23082d9d6b>", line 1, in <module>
psi[:][:500] = b
ValueError: could not broadcast input array from shape (3,500) into shape (3,1000)
我对主题也有一些变化，但结果相似。这看起来非常简单。知道怎么做吗？