我正在繪制一個圖表來可視化一些排序算法在不同數(shù)據(jù)大小上的運行時間。條件是運行時間應該在 y 軸上，數(shù)據(jù)大小應該在 x 軸上。我通過將不同算法的運行時間與數(shù)據(jù)大小相比較并給每個標記不同的顏色來繪制散點圖。同樣，我繪制了 3 個其他數(shù)據(jù)大小，但對于不同的算法使用相同的顏色。我想在圖表中添加一個圖例，以便用戶了解這個特定的顏色點對應于這個特定的算法。我想不出合適的方法。我在網(wǎng)上搜索了一些場景，他們正在根據(jù)不同的散點圖添加圖例。但是，我想為基于顏色的點添加圖例。此外，對于這種情況，您能否建議更好的繪圖曲線。這是我用于圖形生成的代碼。def visualize_datasize(dataset): datasize=len(dataset) for i in range(4,0,-1): run_time=getRunTime(dataset,int(datasize/i)) plt.scatter([int(datasize/i)]*5,run_time,color=['red','green','blue','yellow','black']) plt.xlabel('Size of the dataset') plt.ylabel('Run time') plt.title('Run time vs datasize for various sorting algorithms') plt.show()
1 回答
慕哥6287543
TA貢獻1831條經(jīng)驗 獲得超10個贊
最簡單、最一致的方法是為每個算法制作散點圖。此外，您可能希望為此使用面向對象的接口。
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
def calculate_runtimes(algo, data, sizes):
if algo == 'name1':
# return timings for algorithm 1 given data at given sizes
elif algo == 'name2':
# ...
algo_labels = ['name1', 'name2', 'name3', 'name4', 'name5']
sizes = [1, 2, 4, 8, 16]
algo_runtimes = {name: calculate_runtimes(name, dataset, sizes) for name in algo_labels}
colors = ['red', 'green', 'blue', 'yellow', 'black']
x_positions = [len(dataset)*size for size in sizes]
for (label, runtimes), color in zip(algo_runtimes.items(), colors):
ax.scatter(x_positions, runtimes, color=color, label=label)
ax.legend()
添加回答
舉報
0/150
提交
取消