Непрерывная гистограмма Цвет с плотно - PullRequest
0 голосов
/ 18 апреля 2020

Я создал график, показывающий мне координаты случайного блуждания в виде гистограмм. Я хочу добавить атрибут в функцию plot, чтобы гистограммы отображались непрерывным цветом от светло-голубого (первая гистограмма в списке) до темно-синего (последняя гистограмма в списке). Есть ли не замысловатый способ сделать это?

from plotly.graph_objs import Bar, Layout
from plotly import offline

from random_walk import RandomWalk


# Make a random walk.
rw = RandomWalk(50)
rw.fill_walk()

x_values = rw.x_values
y_values = rw.y_values

# Visualize the results.
data = [Bar(x=x_values, y=y_values)]

x_axis_config = {'title': 'Horizontal Coordinate'}
y_axis_config = {'title': 'Vertical Coordinate'}
my_layout = Layout(title='Results of a 1000 Steps Random Walk',
    xaxis=x_axis_config, yaxis=y_axis_config)
offline.plot({'data': data, 'layout': my_layout}, 
    filename='rw_visual_plotly.html')

класс для создания шагов:

from random import choice

class RandomWalk:
    """A class to generate random walks."""

    def __init__(self, num_points=5000):
        """Initialize attributes of a walk."""
        self.num_points = num_points

        # All walks start at (0, 0).
        self.x_values = [0]
        self.y_values = [0]

    def fill_walk(self):
        """Calculate all the points in the walk."""

        # Keep taking steps until the walk reaches the desired length.
        while len(self.x_values) < self.num_points:

            # Decide which direction to go and how far to go in that direction.
            x_direction = choice([-1, 1])
            x_distance = choice([0, 1, 2, 3, 4])
            x_step = x_direction * x_distance

            y_direction = choice([-1, 1])
            y_distance = choice([0, 1, 2, 3, 4])
            y_step = y_direction * y_distance

            # Reject moves that go nowhere.
            if x_step == 0 and y_step == 0:
                continue

            # Calculate the new position.
            x = self.x_values[-1] + x_step
            y = self.y_values[-1] + y_step

            self.x_values.append(x)
            self.y_values.append(y)    

Ответы [ 2 ]

1 голос
/ 19 апреля 2020

См. это сообщение на форуме сообщества Plotly.

from plotly.graph_objs import Bar, Layout
from plotly import offline

# Make a random walk.
rw = RandomWalk(50)
rw.fill_walk()

x_values = rw.x_values
y_values = rw.y_values

# Visualize the results.
data = [Bar(x=x_values, y=y_values, marker={'color': x_values, 'colorscale': 'Blues'})]

x_axis_config = {'title': 'Horizontal Coordinate', 'linecolor': 'gray', 'mirror': True}
y_axis_config = {'title': 'Vertical Coordinate', 'linecolor': 'gray', 'mirror': True}

my_layout = Layout(title='Results of a 1000 Steps Random Walk', plot_bgcolor='white',
                   xaxis=x_axis_config, yaxis=y_axis_config)

offline.plot({'data': data, 'layout': my_layout}, filename='rw_visual_plotly.html')

enter image description here

1 голос
/ 18 апреля 2020

Вы можете использовать функцию range_to() в библиотеке color для генерации диапазона цветов между двумя разными цветами (например, светло-синий и темно-синий).

from plotly.graph_objs import Bar, Layout
from plotly import offline
from colour import Color
import numpy as np

# Make a random walk.
rw = RandomWalk(50)
rw.fill_walk()

x_values = np.array(rw.x_values)
y_values = np.array(rw.y_values)

x_values = x_values[np.argsort(x_values)]
y_values = y_values[np.argsort(x_values)]

# Visualize the results.
start_color = '#b2d8ff' # light blue
end_color = '#00264c' # dark blue

# list of colors between 'start_color' and 'end_color'
colorscale = [x.hex for x in list(Color(start_color).range_to(Color(end_color), 50))]

data = []

for i in range(len(x_values)):

    data.append(Bar(x=[x_values[i]], y=[y_values[i]], marker=dict(color=colorscale[i])))

x_axis_config = {'title': 'Horizontal Coordinate'}
y_axis_config = {'title': 'Vertical Coordinate'}

my_layout = Layout(title='Results of a 1000 Steps Random Walk', 
                   xaxis=x_axis_config, yaxis=y_axis_config, 
                   showlegend=False, barmode='stack')

offline.plot({'data': data, 'layout': my_layout}, filename='rw_visual_plotly.html')

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...