turns-00026.parquet:47437
fcbc404edbc144aedda4ed43degenerate_repetitionAbsentFinal dense release
USER
Выдает ошибку TypeError: alpha must be numeric or None, not <class 'numpy.ndarray'>
ASSISTANT
Поскольку `plot_surface` не принимает массив значений альфа (прозрачности) напрямую, нам потребуется другой подход для отображения каждой точки поверхности с различной прозрачностью.
Решение заключается в использовании метода `Poly3DCollection` для создания множества полигонов с различными значениями прозрачности. Давайте проведем соответствующие изменения в вашем коде.
Для этого создадим функцию `plot_surface_with_alpha` для построения поверхности с изменяющейся прозрачностью. Мы будем вычислять цвет и прозрачность для каждого многоугольника отдельно:
```python
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
def calculate_alpha(z_value, F_min, F_max):
min_alpha = 0.2
max_alpha = 1.0
return min_alpha + (max_alpha - min_alpha) * (z_value - F_min) / (F_max - F_min)
def plot_surface_with_alpha(ax, I, V, F, facecolors, alpha):
F_min = F.min()
F_max = F.max()
polygons = []
colors = []
for i in range(len(I) - 1):
for j in range(len(V) - 1):
z_value = F[i, j]
a = calculate_alpha(z_value, F_min, F_max)
polygon = [(I[i, j], V[i, j], F[i, j]),
(I[i, j + 1], V[i, j + 1], F[i, j + 1]),
(I[i + 1, j + 1], V[i + 1, j + 1], F[i + 1, j + 1]),
(I[i + 1, j], V[i + 1, j], F[i + 1, j])]
polygons.append(polygon)
colors.append((*facecolors(i, j)[:3], a))
poly3dcollection = Poly3DCollection(polygons, facecolors=colors, edgecolor='none')
ax.add_collection3d(poly3dcollection)
def graph_view(ax, sigmaI, muI, sigmaV, muV, Imin, Imax, Vmin, Vmax, alpha, colorPart, color, alphaPart):
I, V = np.meshgrid(np.linspace(IlimMin, IlimMax, 150), np.linspace(VlimMin, VlimMax, 150))
F = np.exp((-(I-muI)**2-(np.log(V)-muV)**2)/(2 * sigmaI**2 * sigmaV**2))/(sigmaI * sigmaV * V * 2*np.pi)
plot_surface_with_alpha(ax, I, V, F, plt.cm.viridis, alpha)
mask = (I >= Imin) & (I <= Imax) & (V >= Vmin) & (V <= Vmax)
x_highlight = np.copy(I)
y_highlight = np.copy(V)
z_highlight = np.copy(F)
x_highlight[~mask] = np.nan
y_highlight[~mask] = np.nan
z_highlight[~mask] = np.nan
plot_surface_with_alpha(ax, x_highlight, y_highlight, z_highlight, plt.cm.viridis, alphaPart)
ax.set_xlabel('Потребляемый ток')
ax.set_ylabel('Скорость')
ax.set_zlabel('Плотность веростности')
return ax
def graph_view_without_part(ax, sigmaI, muI, sigmaV, muV, alpha, color):
I, V = np.meshgrid(np.linspace(IlimMin, IlimMax, 150), np.linspace(VlimMin, VlimMax, 150))
F1 = np.exp(-0.5*((I-muI)/sigmaI)**2)
F2 = np.exp(-0.5*((V-muV)/sigmaV)**2)
F = F1*F2
plot_surface_with_alpha(ax, I, V, F, plt.cm.viridis, alpha)
max_idx = np.unravel_index(np.argmax(F, axis=None), F.shape)
max_I = I[max_idx]
max_V = V[max_idx]
max_F = F[max_idx]
print(f"Max I: {max_I}, Max V: {max_V}, Max F: {max_F}")
ax.set_xlabel('Потребляемый ток')
ax.set_ylabel('Скорость')
ax.set_zlabel('Плотность веростности')
return ax
def background(ax):
zmin = 0
square1 = np.array([[0, 0, zmin], [3.33, 0, zmin], [3.33, 0.75, zmin], [0, 0.75, zmin]])
square2 = np.array([[3.33, 0, zmin], [10, 0, zmin], [10, 0.75, zmin], [3.33, 0.75, zmin]])
square3 = np.array([[10, 0, zmin], [16.66, 0, zmin], [16.66, 0.75, zmin], [10, 0.75, zmin]])
square4 = np.array([[16.66, 0, zmin], [20, 0, zmin], [20, 0.75, zmin], [16.66, 0.75, zmin]])
square5 = np.array([[3.33, 0.75, zmin], [10, 0.75, zmin], [10, 2.25, zmin], [3.33, 2.25, zmin]])
square6 = np.array([[10, 0.75, zmin], [16.66, 0.75, zmin], [16.66, 2.25, zmin], [10, 2.25, zmin]])
square7 = np.array([[16.66, 0.75, zmin], [20, 0.75, zmin], [20, 2.25, zmin], [16.66, 2.25, zmin]])
square8 = np.array([[10, 2.25, zmin], [16.66, 2.25, zmin], [16.66, 3.75, zmin], [10, 3.75, zmin]])
square9 = np.array([[16.66, 2.25, zmin], [20, 2.25, zmin], [20, 3.75, zmin], [16.66, 3.75, zmin]])
square10 = np.array([[10, 3.75, zmin], [16.66, 3.75, zmin], [16.66, 4.5, zmin], [10, 4.5, zmin]])
square11 = np.array([[16.66, 3.75, zmin], [20, 3.75, zmin], [20, 4.5, zmin], [16.66, 4.5, zmin]])
sq = [square1, square2, square3, square4, square5, square6, square7, square8, square9, square10, square11]
colors = ['green', 'gold', 'red', 'firebrick', 'gold', 'gold', 'firebrick', 'gold', 'red', 'red', 'red']
for i in range(len(sq)):
square = Poly3DCollection([sq[i]], color=colors[i], edgecolor='none', alpha = 0.5)
ax.add_collection3d(square)
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, projection='3d', computed_zorder=False)
IlimMin = 0
IlimMax = 20
VlimMin = 0
VlimMax = 4.5
sigmaI = 2.33
muI = 11.46
sigmaV = 0.755
muV = -0.611
Imin, Imax = 17.04, 19.67
Vmin, Vmax = 0.55, 0.74
SecSurf = graph_view(ax, sigmaI, muI, sigmaV, muV, Imin, Imax, Vmin, Vmax, alpha=0.8, alphaPart=0.6, colorPart='salmon', color='green')
sigmaI = 2.398
muI = 10.32
sigmaV = 0.75
muV = -0.625
Imin, Imax = 16.35, 16.55
Vmin, Vmax = 0.359, 0.529
ThirdSurf = graph_view(ax, sigmaI, muI, sigmaV, muV, Imin, Imax, Vmin, Vmax, alpha=0.7, alphaPart=1, colorPart='navy', color='violet')
sigmaI = 1.7
muI = 12.8
sigmaV = 0.71
muV = -0.554
Imin, Imax = 17.04, 17.49
Vmin, Vmax = 0.6, 0.74
firstSurf = graph_view(ax, sigmaI, muI, sigmaV, muV, Imin, Imax, Vmin, Vmax, alpha=0.8, alphaPart=1, colorPart='firebrick', color='royalblue')
sigmaI = 1.7
muI = 18
sigmaV = 0.71
muV = 0.9
modelSurf = graph_view_without_part(ax, sigmaI, muI, sigmaV, muV, alpha=0.4, color='tomato')
firstSurf.set_zorder(4)
SecSurf.set_zorder(3)
ThirdSurf.set_zorder(2)
modelSurf.set_zorder(5)
background(ax)
plt.show()
```
В этой версии кода мы определяем функцию `plot_surface_with_alpha`, которая создает множество полигонов с измененной альфа-прозрачностью, зависящей от высоты (значения по оси Z). Мы используем эту функцию вместо `plot_surface` внутри функций `graph_view` и `graph_view_without_part`, что позволяет отобразить поверхности с плавно изменяющейся прозрачностью.