Продолжительность сюжета в матплотлибе / морском роге - PullRequest
Новая
       13

Продолжительность сюжета в матплотлибе / морском роге

0 голосов
/ 09 июня 2019

Я хочу построить результаты пробега 5K и 10K в Python с matplotlib и seaborn.В моем наборе данных есть столбец time, который содержит строковый объект в формате HH:MM:SS, например 00:28:50 или 1:17:23, с результатами гонки.Я создал свой график, рассчитав время в секундах, но я предпочитаю фактическое время в формате HH:MM:SS на оси для удобства чтения.

Любые предложения, чтобы сделать это?

Мой код такдалеко (с фиктивными данными): 

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

df = pd.DataFrame({'sex': ['M', 'M', 'M', 'M', 'F', 'F', 'F', 'F'], 'race': ['5K', '5K', '10K', '10K', '5K', '5K', '10K', '10K'], 'time': ['00:20:16', '00:24:57', '00:49:17', '00:56:10', '00:26:31', '00:33:06', '00:58:29', '01:05:03']})

df['time'] = pd.to_datetime(df['time'])
df['time_sec'] =[(t.hour * 3600 + t.minute * 60 + t.second) for t in df.time]
order=['5K', '10K']
palette = ['#3498db', '#ff0080']
fig, ax = plt.subplots(figsize=(16, 8))
sns.boxplot(ax=ax, data=df, x='time_sec', y='race', hue='sex', order=order, palette=palette, orient='h', linewidth=2.5)
plt.title('Time in seconds', fontsize=16)
plt.show()

enter image description here

1 Ответ

1 голос
/ 10 июня 2019

Одна вещь, которую вы можете сделать, это изменить тики вручную: 

order=['5K', '10K']
palette = ['#3498db', '#ff0080']
fig, ax = plt.subplots(figsize=(16, 8))
sns.boxplot(ax=ax, data=df, x='time_sec', y='race', hue='sex', order=order, palette=palette, orient='h', linewidth=2.5)

# get the ticks 
ticks = ax.get_xticks()

# convert the ticks to string
ax.set_xticklabels(pd.to_datetime(ticks, unit='s').strftime('%H:%M:%S'))

plt.title('Time in seconds', fontsize=16)

plt.show()

Вывод:

enter image description here

...