如何处理函数不连续性以消除PyVista等值面分支伪影？

问题根源在于np.angle返回的角度范围是[-π, π]，当角度从π跳变到-π时，网格相邻点间会产生虚假的大数值差，导致contour插值时出现分支点伪影。

核心解决方案 #

使用np.unwrap函数将角度展开为连续的数值序列，消除±π处的不连续性。该函数会自动检测并修正角度的跳变，将其转换为连续的缠绕角（比如把-π调整为π+2π，保证相邻点角度变化平滑）。

修改后的代码 #

import numpy as np
import pyvista as pv

a = 2
b = 3
theta = 1

def toComplex(x, y, z):
    z1 = 2*x + 2*y*1j
    z2 = 2*z + (x*x + y*y + z*z - 1)*1j
    z1 /= x*x + y*y + z*z + 1
    z2 /= x*x + y*y + z*z + 1
    return [z1, z2]

#%% Data
x, y, z = np.pi*np.mgrid[-2:2:101j, -2:2:101j, -2:2:101j]
complexPoint = toComplex(x, y, z)
argument = np.angle(complexPoint[0]**2 + complexPoint[1]**3)

# 关键步骤：沿所有维度展开角度，消除不连续性
argument_unwrapped = np.unwrap(np.unwrap(np.unwrap(argument, axis=0), axis=1), axis=2)

# 选择对应连续角度的等值面值：原theta=1，可根据需要调整周期数
# 比如如果unwrap后的角度包含多个周期，可使用 theta + 2*np.pi*n（n为整数）
contour_value = theta

vol = argument_unwrapped
grid = pv.StructuredGrid(x, y, z)
grid["vol"] = vol.flatten()
contours = grid.contour([contour_value])

#%% Visualization
pv.set_plot_theme('default')
p = pv.Plotter()
p.add_mesh(contours, scalars=contours.points[:, 2], show_scalar_bar=False)
p.show(interactive_update=False)

说明 #

1. np.unwrap需要沿所有维度（x、y、z对应axis=0、1、2）处理，确保整个3D网格中的角度都连续。
2. 展开后的角度范围不再局限于[-π, π]，可能会延伸到[-π+2πk, π+2πk]（k为整数）。因此等值面的contour_value需要对应调整：如果原theta在[-π, π]，可直接使用原数值，或根据需要加上2π的整数倍来获取不同周期的等值面。
3. 处理后，网格中不再存在π到-π的跳变，contour插值时会生成平滑的等值面，消除分支点伪影。

内容的提问来源于stack exchange，提问作者Eduardo de Lorenzo