Matplotlib Quiver colormap ⇐ Python

[Anonymous]

Я пытаюсь построить интерполированные данные о скорости из -за свободы в Python. Интерполированные векторные компоненты скорости нормализуются по размеру и построены с помощью функции дрожания. Я пытаюсь представлять величину скорости с коллегой, но что бы я ни делал, стрелы черные. Я что -то упускаю?import pandas as pd
import numpy as np
import scipy
import matplotlib.pyplot as plt
import matplotlib

input_filename = 'period_rot_shadow_vel-mag-vel_comps.txt'

df_results = pd.read_csv(input_filename,
sep =',', skiprows = 1,
names = ['nodenumber', 'X', 'Y', 'Z', 'vel_mag', 'v_x', 'v_y', 'v_z',
'v_ax', 'v_rad', 'v_tang'])

x_data_min = df_results['X'].min()
x_data_max = df_results['X'].max()
y_data_min = df_results['Y'].min()
y_data_max = df_results['Y'].max()

x_size = np.abs(x_data_max) + np.abs(x_data_min)
y_size = np.abs(y_data_max) + np.abs(y_data_min)

rect_mesh_spacing = 0.001

N_x_steps = int(x_size/rect_mesh_spacing) + 1
N_y_steps = int(y_size/rect_mesh_spacing) + 1

rect_mesh_x_axis = np.array([])
rect_mesh_y_axis = np.array([])

for i in range(0, N_x_steps):
x = x_data_min + i * rect_mesh_spacing
rect_mesh_x_axis = np.append(rect_mesh_x_axis, x)

for i in range(0, N_y_steps):
y = y_data_min + i * rect_mesh_spacing
rect_mesh_y_axis = np.append(rect_mesh_y_axis, y)

#use meshgrid function to create a rectangular grid, meshgrid returns a tuple of ndarrays
xx, yy = np.meshgrid(rect_mesh_x_axis, rect_mesh_y_axis)

#create an array from meshgrid points
xy_mesh = np.array((xx,yy)).T

#define input coordinates (imporeted data point coordinates) as numpy arrays
input_coords = np.array(list(zip(df_results['X'].to_numpy(), df_results['Y'].to_numpy())))

# scipy.spatial.KDTree function is used to lookup nearest-neighbour for defined points
tree = scipy.spatial.KDTree(input_coords)

dist, idx = tree.query(xy_mesh)
#define the radius of lookup of vicinity of points
radius = (rect_mesh_spacing + rect_mesh_spacing) / 2
distance_filter = dist

Подробнее здесь: https://stackoverflow.com/questions/796 ... r-colormap