scatter_squares.py
import matplotlib.pyplot as plt
x_values = list(range(1001))
y_values = [x**2 for x in x_values]
ax.scatter(x_values, y_values, c=y_values, cmap=plt.cm.Blues, s=10)
# Назначение заголовка диаграммы и меток осей.
...
Построение простого графика
329
Мы передаем в
c
список значений по оси
y
, а затем указываем
pyplot
, какая цветовая
карта должна использоваться, при помощи аргумента
cmap
. Следующий код окра-
шивает точки с меньшими значениями
y
в светло-синий цвет, а точки с большими
значениями
y
— в темно-синий цвет. Полученная диаграмма изображена на рис. 15.8.
Рис. 15.8.
Точечная диаграмма с цветовой картой Blues
ПРИМЕЧАНИЕ
Все цветовые карты, доступные в pyplot, можно просмотреть на сайте
http://matplotlib .org/
; откройте раздел
Examples
, прокрутите содержимое до пункта
Color
и щелкните на ссылке
Colormaps_reference
.
Автоматическое сохранение диаграмм
Если вы хотите, чтобы программа автоматически сохраняла диаграмму в файле,
замените вызов
plt.show()
вызовом
plt.savefig()
:
plt.savefig('squares_plot.png', bbox_inches='tight')
Первый аргумент содержит имя файла для сохранения диаграммы; файл будет
сохранен в одном каталоге с
scatter_squares .py
. Второй аргумент отсекает от диа-
граммы лишнее пространство. Если вы хотите оставить пустые места вокруг диа-
граммы, этот аргумент можно опустить.
УПРАЖНЕНИЯ
15.1. Кубы:
число, возведенное в третью степень, называется «кубом». Нанесите на диа-
грамму первые пять кубов, а затем первые 5000 кубов.
15.2. Цветные кубы:
примените цветовую карту к диаграмме с кубами.
330
Достарыңызбен бөлісу: |