колчан matplotlib: неверный аргумент RGBA - PullRequest
Новая
       36

колчан matplotlib: неверный аргумент RGBA

0 голосов
/ 03 апреля 2020

Я написал этот код 

from scipy.ndimage import gaussian_filter

U = gaussian_filter(optflow[1][:,:,0], sigma = 10)
V = gaussian_filter(optflow[1][:,:,1], sigma = 10)
U[abs(U) < 0.5] = 0
V[abs(V) < 0.5] = 0

plt.quiver(U[::7, ::7], V[::7, ::7], scale=0.4, scale_units = 'xy', headlength = 7)

#plt.axis('off')
plt.gca().invert_yaxis()
plt.savefig('/home/RR/TEST0.png')
matplotlib.pyplot.clf()

, который строит векторное поле этой формы:

Мой вопрос: как я могу изменить цвет стрелок в соответствии с их модулем?

Я пробовал это: 

%matplotlib inline
%pylab
from scipy.ndimage import gaussian_filter

U = gaussian_filter(optflow[1][:,:,0], sigma = 10)
V = gaussian_filter(optflow[1][:,:,1], sigma = 10)
U[abs(U) < 0.5] = 0
V[abs(V) < 0.5] = 0

M = np.sqrt(U*U+V*V)

plt.quiver(U[::7, ::7], V[::7, ::7], color = cm.inferno(M), scale=0.4, scale_units = 'xy', headlength = 7)

#plt.axis('off')
plt.gca().invert_yaxis()
plt.savefig('/home/RR7g.png')
matplotlib.pyplot.clf()

, но появляется следующая ошибка. 

Invalid RGBA argument

1 Ответ

1 голос
/ 03 апреля 2020 
%matplotlib inline
%pylab
from scipy.ndimage import gaussian_filter
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors

U = gaussian_filter(optflow[1][:,:,0], sigma = 10)
V = gaussian_filter(optflow[1][:,:,1], sigma = 10)
U[abs(U) < 0.5] = 0
V[abs(V) < 0.5] = 0

M = np.sqrt(U*U+V*V)
# cmap = plt.cm.get_cmap("inferno", 7).colors
# colors = list((mcolors.to_hex(x) for x in cmap))
plt.quiver(U[::7, ::7], V[::7, ::7], C=M, cmap='inferno', scale=0.4, scale_units = 'xy', headlength = 7)

#plt.axis('off')
plt.gca().invert_yaxis()
plt.savefig('/home/roberto/workspace/TEST/OUTPUT/CORRECTIONCLEAN/TEST0.png')
matplotlib.pyplot.clf()

EDIT опускает скобку EDIT 2 измененного генератора в список EDIT 3, пытаясь установить цвет на величину стрелки

...