我必须通过创建循环来使用matplotlib绘制多条线和标记,并且我已经在matplolibrc参数文件中设置了轴的颜色循环。在循环的每个循环中,都会创建一组标记和线条(线条由单独的命令创建)。但是,标记和线的颜色随轴颜色周期而不同。我希望每次运行循环时,标记和线条都具有与该循环相同的颜色。
我已经包含了可复制的代码:
import numpy as np
import itertools
import matplotlib.pyplot as plt
m = 5
n = 5
x = np.zeros(shape=(m, n))
plt.figure(figsize=(5.15, 5.15))
plt.clf()
plt.subplot(111)
marker = itertools.cycle(('o', 'v', '^', '<', '>', 's', '8', 'p'))
for i in range(1, n):
x = np.dot(i, [1, 1.1, 1.2, 1.3])
y = x ** 2
plt.plot(x, y, linestyle='', markeredgecolor='none', marker=marker.next())
plt.plot(x, y, linestyle='-')
plt.ylabel(r'\textit{y}', labelpad=6)
plt.xlabel(r'\textit{x}', labelpad=6)
plt.show()
使用此代码,我得到的输出是:
我需要在其范围内绘制的标记和线条具有相同的颜色。如何在matplotlib中完成?
我介绍两种方式。第一个可能更干净:它循环一次,并且在每个循环中获取下一种颜色,然后使用该颜色执行两个绘图命令。在第二个步骤中,它循环并执行所有标记,然后重置颜色并再次循环并执行线条。
第一种方法
直接访问颜色循环。如果是matplotlib 1.5或更高版本,请使用color=next(ax._get_lines.prop_cycler)['color']
。否则color=next(ax._get_lines.color_cycle)
:
import numpy as np
import itertools
import matplotlib.pyplot as plt
m = 5
n = 5
x = np.zeros(shape=(m, n))
plt.figure(figsize=(5.15, 5.15))
plt.clf()
plt.subplot(111)
marker = itertools.cycle(('o', 'v', '^', '<', '>', 's', '8', 'p'))
ax = plt.gca()
for i in range(1, n):
x = np.dot(i, [1, 1.1, 1.2, 1.3])
y = x ** 2
#
#for matplotlib before 1.5, use
#color = next(ax._get_lines.color_cycle)
#instead of next line (thanks to Jon Loveday for the update)
#
color = next(ax._get_lines.prop_cycler)['color']
plt.plot(x, y, linestyle='', markeredgecolor='none', marker=marker.next(), color=color)
plt.plot(x, y, linestyle='-', color = color)
plt.ylabel(r'$y$', labelpad=6)
plt.xlabel(r'$x$', labelpad=6)
plt.savefig('test2.png')
另外,请注意,我将您更改\textit{y}
为$y$
。通常,您实际上实际上需要数学字体而不是斜体。
第二种方法
如果您不想将颜色作为参数发送,则可以重置颜色循环并循环两次。在这里,我将其在第一个循环之前重置,然后在第二个循环之前再次重置,以确保它从同一位置开始。
import numpy as np
import itertools
import matplotlib.pyplot as plt
m = 5
n = 5
x = np.zeros(shape=(m, n))
plt.figure(figsize=(5.15, 5.15))
plt.clf()
plt.subplot(111)
marker = itertools.cycle(('o', 'v', '^', '<', '>', 's', '8', 'p'))
plt.gca().set_prop_cycle(None) #if matplotlib <1.5 use set_color_cycle
for i in range(1, n):
x = np.dot(i, [1, 1.1, 1.2, 1.3])
y = x ** 2
plt.plot(x, y, linestyle='', markeredgecolor='none', marker=marker.next())
plt.gca().set_prop_cycle(None)
for i in range(1, n):
x = np.dot(i, [1, 1.1, 1.2, 1.3])
y = x ** 2
plt.plot(x, y, linestyle='-')
plt.ylabel(r'$y$', labelpad=6)
plt.xlabel(r'$x$', labelpad=6)
plt.savefig('test.png')
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句