在 12 小时格式的 Python 绘图时间中修复了 y 轴

Question

我有这个图，但我需要将 y 轴固定为 00:00、01:00、02:00，一直到 12:00。截至目前，它只是在 y 轴上绘制我在 csv 中的值。 csv 格式如下。如何让 y 轴保持不变，仅以 1 小时的增量显示 00:00 到 12:00，并且仍然正确绘制数据？

    ML  INT 0.1     534.15  0:00
    ML  EXT 0.25    654.23  3:00
    ML  INT 0.35    743.12  6:30

下面是我目前的代码。

    import pandas as pd
    import matplotlib.pyplot as plt
    import numpy as np

    data = pd.read_csv('data.csv', header=None)
    ints = data[data[1]=='INT']
    exts = data[data[1]=='EXT']
    INT_index = data[data[1]=='INT'].index
    EXT_index = data[data[1]=='EXT'].index
    time = [t for t in data[4]]
    int_dist = [d for d in ints[3]]
    ext_dist = [d for d in exts[3]]


    fig, ax = plt.subplots()
    ax.scatter(int_dist, INT_index, c='orange', s=150)
    ax.scatter(ext_dist, EXT_index, c='black', s=150)
    ax.set_yticks(np.arange(len(data[4])))
    ax.set_yticklabels(time)
    plt.legend(['INT', 'EXT'], loc=4)
    plt.xlabel('Distance')
    plt.ylabel('Time')
    plt.show()

Answer 1

我生成了更多行数据来使问题更有意义，至少对我而言是这样。

为我解决这个问题的是生成第 5 列（在代码中，而不是 csv 中），它是对应于特定点钟时间的分钟数，即 11:59 映射到 719 分钟。使用 pandas 我将这个新列插入到数据框中。然后我可以每 60 分钟放置一个每小时的字符串标记标签（'0:00'、'1:00' 等）。

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

data = pd.read_csv('Workbook2.csv', header=None)
print data

打印我伪造的数据：

    0    1     2       3      4
0  ML  INT  0.10  534.15   0:00
1  ML  EXT  0.25  654.23   3:00
2  ML  INT  0.30  743.12   6:30
3  ML  EXT  0.35  744.20   4:30
4  ML  INT  0.45  811.47   7:00
5  ML  EXT  0.55  777.90   5:45
6  ML  INT  0.66  854.70   7:54
7  ML  EXT  0.74  798.40   6:55
8  ML  INT  0.87  947.30  11:59

现在创建一个将点钟转换为分钟的函数：

def convert_to_min(o_clock):
    h, m = o_clock.split(':')
    return int(h) * 60 + int(m)
# using this function create a list times in minutes for each time in col 4
min_col = [convert_to_min(t) for t in data[4]]
data[5] = min_col  # inserts this list as a new column '5'
print data

我们的新数据：

    0    1     2       3      4    5
0  ML  INT  0.10  534.15   0:00    0
1  ML  EXT  0.25  654.23   3:00  180
2  ML  INT  0.30  743.12   6:30  390
3  ML  EXT  0.35  744.20   4:30  270
4  ML  INT  0.45  811.47   7:00  420
5  ML  EXT  0.55  777.90   5:45  345
6  ML  INT  0.66  854.70   7:54  474
7  ML  EXT  0.74  798.40   6:55  415
8  ML  INT  0.87  947.30  11:59  719

现在构建 x 和 y 轴数据、刻度标签和刻度位置：

INTs = data[data[1]=='INT']
EXTs = data[data[1]=='EXT']

int_dist = INTs[3]  # x-axis data for INT
ext_dist = EXTs[3]

# plotting time as minutes in range [0 720]
int_time = INTs[5]  # y-axis data for INT
ext_time = EXTs[5]

time = ['0:00', '1:00', '2:00', '3:00', '4:00', '5:00', 
        '6:00', '7:00', '8:00', '9:00', '10:00', '11:00', '12:00']
# this will place the strings above at every 60 min
tick_location = [t*60 for t in range(13)]

现在剧情：

fig, ax = plt.subplots()
ax.scatter(int_dist, int_time, c='orange', s=150)
ax.scatter(ext_dist, ext_time, c='black', s=150)
ax.set_yticks(tick_location)
ax.set_yticklabels(time)
plt.legend(['INT', 'EXT'], loc=4)
plt.xlabel('Distance')
plt.ylabel('Time')
plt.title('Seems to work...')
plt.show()

Answer 2

如果您使用 datetime 作为 y 轴，刻度会更智能。

假数据：

df = pd.DataFrame({'value':[530,640,710], 'time':['0:00', '3:00', '6:30']})

   time  value
0  0:00    530
1  3:00    640
2  6:30    710

将 df.time 从 str 转换为 datetime:

time2 = pd.to_datetime(df.time, format='%H:%M')
plt.plot(df.value, time2, marker='o', linestyle='None')

似乎无法将其放入 scatter 而不是 plot 以防它对您很重要（我取消了该行）。可能是因为 datetime 应该始终在时间序列线图中而不是在散点图中（我欢迎评论让我知道是否确实如此并且 datetime 不能放入 scatter） .

在 12 小时格式的 Python 绘图时间中修复了 y 轴

Fixed y axis in Python plotting times in 12 hr format

python

time

matplotlib

scatter