【Python】学会它，谁还用循环嵌套

所谓循环嵌套，就是这种

for i in range(5):
	for j in range(5):
		for k in range(5):
			print(i,j,k)

对于有代码洁癖的人显然是无法忍受的。

现在，你有了更好的选择，调用itertools中的product，可以更加优雅地完成这个工作

for i,j,k in product(range(5),range(5),range(5)):
	print(i,j,k)

用这个方法可以画一个点阵出来

from itertools import product
import matplotlib.pyplot as plt

ax = plt.subplot(projection='3d')
col = range(5)
for r,g,b in product(col,col,col):
	ax.scatter([r],[g],[b],c=(r/5,g/5,b/5))
plt.show()

或者用把所有的点放在一起画

col = [i/32 for i in range(32)]
colors = list(product(col,col,col))
rs,gs,bs = zip(*colors)
ax = plt.subplot(projection='3d')
ax.scatter(rs,gs,bs,c=colors)
plt.show()

一共有三万多点，可能有点慢