python addition problem ????

最新推荐文章于 2022-07-11 14:23:45 发布

Whity_Yuan

最新推荐文章于 2022-07-11 14:23:45 发布

阅读量1.4k

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/Whity_Yuan/article/details/85435993

版权

Python 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

These 2 blocks of code give different result, why?

import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D


def midpoints(x):
    sl = ()
    for i in range(x.ndim):
        x = (x[sl + np.index_exp[:-1]] + x[sl + np.index_exp[1:]]) / 2.0
        sl += np.index_exp[:]
    return x

# prepare some coordinates, and attach rgb values to each
r, g, b = np.indices((5, 5, 5)) / 4.0
rc = midpoints(r)
gc = midpoints(g)
bc = midpoints(b)

# define a sphere about [0.5, 0.5, 0.5]
sphere = (rc - 0.5)**2 + (gc - 0.5)**2 + (bc - 0.5)**2 < 0.5**2

# combine the color components
colors = np.zeros(sphere.shape + (3,))
colors[..., 0] = rc
colors[..., 1] = gc
colors[..., 2] = bc

# and plot everything
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.voxels(r, g, b, sphere,
          facecolors=colors,
          edgecolors=np.clip(2*colors - 0.5, 0, 1),  # brighter
          linewidth=0.5)
ax.set(xlabel='r', ylabel='g', zlabel='b')

plt.show()

def midpoints(x):
	sl = ()

	for i in range(x.ndim):
		print(sl + np.index_exp[:-1])
		print(sl + np.index_exp[:1])
		print(x)
		print("===========================")
		a=x[sl + np.index_exp[:-1]]
		print(a)
		print("===========================")
		b=x[sl + np.index_exp[:1]]
		print(b)
		print("===========================")
		x = (a + b) / 2.0
		sl += np.index_exp[:]
		print(x)
	return x

# prepare some coordinates, and attach rgb values to each
r, g, b = np.indices((5, 5, 5)) / 4
#print(np.indices((3, 3, 3)))
rc = midpoints(r)
gc = midpoints(g)
bc = midpoints(b)

# define a sphere about [0.5, 0.5, 0.5]
sphere = (rc - 0.5)**2 + (gc - 0.5)**2 + (bc - 0.5)**2 < 0.5**2

# combine the color components
colors = np.zeros(sphere.shape + (3,))
colors[..., 0] = rc
colors[..., 1] = gc
colors[..., 2] = bc

# and plot everything
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.voxels(r, g, b, sphere,
          facecolors=colors,
          edgecolors=np.clip(2*colors - 0.5, 0, 1),  # brighter
          linewidth=0.5)
ax.set(xlabel='r', ylabel='g', zlabel='b')

plt.show()