ПРИМЕЧАНИЕ
Если у вас версия Python 2 .7, используйте raw_input() вместо input() в точке .
Оформление случайного блуждания
В этом разделе мы настроим диаграмму так, чтобы подчеркнуть важные ха-
рактеристики каждого блуждания и отвести на второй план несущественные
элементы. Для этого мы выделим характеристики, которые нужно подчеркнуть
(например, откуда началось блуждание, где оно закончилось и по какому пути
следовало). Затем определяются характеристики, которые нужно ослабить (на-
пример, деления шкалы и метки). Результатом должно быть простое визуальное
представление, которое четко описывает путь, использованный в каждом слу-
чайном блуждании.
Назначение цветов
Мы используем цветовую карту для отображения точек блуждания, а также уда-
ляем черный контур из каждой точки, чтобы цвет точек был лучше виден. Чтобы
точки окрашивались в соответствии с их позицией в блуждании, мы передаем в ар-
гументе
c
список с позицией каждой точки. Так как точки выводятся по порядку,
список просто содержит числа от 1 до 5000:
rw_visual.py
...
while True:
# Построение случайного блуждания и нанесение точек на диаграмму.
rw = RandomWalk()
rw.fill_walk()
point_numbers = list(range(rw.num_points))
plt.scatter(rw.x_values, rw.y_values, c=point_numbers, cmap=plt.cm.Blues,
edgecolor='none', s=15)
plt.show()
keep_running = input("Make another walk? (y/n): ")
...
В точке функция
range()
используется для генерирования списка чисел, размер
которого равен количеству точек в блуждании. Полученный результат сохраняется
в списке
point_numbers
, который используется для назначения цвета каждой точки
324 Глава 15 • Генерирование данных
в блуждании. Мы передаем
point_numbers
в аргументе
c
, используем цветовую
карту
Blues
и затем передаем
edgecolor=none
для удаления черного контура вокруг
каждой точки. В результате создается диаграмма блуждания с градиентным пере-
ходом от светло-синего к темно-синему (рис. 15.9).
Достарыңызбен бөлісу: |