Mayavi исправил цветовую панель для всех сцен - PullRequest
Новая
       49

Mayavi исправил цветовую панель для всех сцен

0 голосов
/ 31 августа 2018

Я ищу способ сохранить постоянство цветовой шкалы (и цвета на сцене) независимо от того, какие данные я отображаю (например, допустим, я хочу, чтобы цвета были от 0 до 10, с текущей настройкой Mayavi будет корректировать цвета в соответствии с данными. , поэтому первый график будет от 0 до 1, второй от 0 до 2 и т. д.). Нормализация (vmin, vmax) не вариант, поскольку они просто нормализуют данные. 

#Minimum working example
from numpy import sin, cos, mgrid, pi, sqrt
from mayavi import mlab

mlab.figure(fgcolor=(0, 0, 0), bgcolor=(1, 1, 1))
u, v = mgrid[- 0.035:pi:0.01, - 0.035:pi:0.01]

X = 2 / 3. * (cos(u) * cos(2 * v)
    + sqrt(2) * sin(u) * cos(v)) * cos(u) / (sqrt(2) -
                                             sin(2 * u) * sin(3 * v))
Y = 2 / 3. * (cos(u) * sin(2 * v) -
    sqrt(2) * sin(u) * sin(v)) * cos(u) / (sqrt(2)
    - sin(2 * u) * sin(3 * v))
Z = -sqrt(2) * cos(u) * cos(u) / (sqrt(2) - sin(2 * u) * sin(3 * v))

for i in range(1,10):
    S = sin(u)*i
    mlab.mesh(X, Y, Z, scalars=S, colormap='YlGnBu', )
    mlab.view(.0, - 5.0, 4)
    mlab.colorbar(orientation='vertical')
    mlab.show()
    mlab.close(all=True)

Ответы [ 2 ]

0 голосов
/ 06 сентября 2018

Вам нужно зафиксировать диапазон данных вашей луты 

# get the current lut manager
lut_manager = mlab.colorbar(orientation='vertical')
# fix the range
lut_manager.data_range = (0, 10)

Таким образом, шкала всегда будет от нуля до десяти, обычно это может быть проблематично, если у вас нет значений во всем диапазоне (цвета могут быть очень похожими)

0 голосов
/ 31 августа 2018

Используя функцию записи , вы можете увидеть, что вы можете управлять этим, изменив менеджер скалярных справочных таблиц : 

#Minimum working example
from numpy import array, sin, cos, mgrid, pi, sqrt
from mayavi import mlab

mlab.figure(fgcolor=(0, 0, 0), bgcolor=(1, 1, 1))
u, v = mgrid[- 0.035:pi:0.01, - 0.035:pi:0.01]

X = 2 / 3. * (cos(u) * cos(2 * v)
    + sqrt(2) * sin(u) * cos(v)) * cos(u) / (sqrt(2) -
                                             sin(2 * u) * sin(3 * v))
Y = 2 / 3. * (cos(u) * sin(2 * v) -
    sqrt(2) * sin(u) * sin(v)) * cos(u) / (sqrt(2)
    - sin(2 * u) * sin(3 * v))
Z = -sqrt(2) * cos(u) * cos(u) / (sqrt(2) - sin(2 * u) * sin(3 * v))

for i in range(1,10):
    S = sin(u)*i
    _m = mlab.mesh(X, Y, Z, scalars=S, colormap='YlGnBu', )
    mlab.view(.0, - 5.0, 4)
    mlab.colorbar(orientation='vertical')
    _m.module_manager.scalar_lut_manager.use_default_range = False
    _m.module_manager.scalar_lut_manager.data_range = array([-1., 10.])
    _m.module_manager.scalar_lut_manager.scalar_bar.position = array([ 0.01,  0.15])
    _m.module_manager.scalar_lut_manager.scalar_bar.position2 = array([ 0.1,  0.8])
    mlab.show()
    mlab.close(all=True)

кажется, что настройка scalar_bar.position сбрасывает LUT, который я сохранил в коде, но вы можете получить другие значения, если записываете свои действия.

...