Gridspec без использования ":" в параметре?

Question

Мой вопрос довольно прост: я не могу сделать «:» в последнем параметре диаграммы (strecke, strecke2, «Strecke in Metern / s», «S (t) Diagramm», 0, :), потому что он говоритменя, чем: «<» не поддерживается между экземплярами «str» и «int», но я хочу иметь эту диаграмму для обоих столбцов, так что я могу сделать это для обоих столбцов?</p>

Ссылка на файл: https://filehorst.de/download.php?file=ceCnteJq

import codecs
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches # wird nicht benutzt, test
import numpy as np
import matplotlib.gridspec as grid

zeit = []
strecke = []
geschwindigkeit = []
beschleunigung = []
strecke2 = []
geschwindigkeit2 = []
beschleunigung2 = []


gs = grid.GridSpec(2, 2)
fig = plt.figure('Diagramme', tight_layout=True)

daten = open("BewegungBeschleunigung.csv")

reiheÜberspringen = True

for i in daten:

    if reiheÜberspringen == True:
        reiheÜberspringen = False
        continue

    i = i.strip().split(",")

    zeit.append(float(i[0]))
    strecke.append(float(i[1]))
    geschwindigkeit.append(float(i[2]))
    beschleunigung.append(float(i[3]))
    strecke2.append(float(i[4]))
    geschwindigkeit2.append(float(i[5]))
    beschleunigung2.append(float(i[6]))

def diagram(y, y2, yachse, titel, pos1, pos2):

    ax = fig.add_subplot(gs[pos1, pos2])
    ax.plot(zeit, y, "go", markersize = 2.5)
    ax.plot(zeit, y2, "ro", markersize = 2.5)
    ax.autoscale(enable = False, axis = "both", tight = None)
    ax.grid(True)
    ax.set_title(titel)

    plt.gcf().canvas.set_window_title("Diagramme")


diagram(strecke, strecke2, "Strecke in Metern/s", "S(t) Diagramm", 0, :)
diagram(geschwindigkeit, geschwindigkeit2, "Geschwindigkeit in Metern/s", "V(t) Diagramm", 1, 0)
diagram(beschleunigung, beschleunigung2, "Beschleunigung in Metern/s^2", "a(t) Diagramm", 1, 1 )
plt.show()



daten.close

Answer 1

Numpy предоставляет способ создания выражения индексации с помощью numpy.s_.Вы можете использовать его как

np.s_[:]

для передачи полного среза в функцию.

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

gs = gridspec.GridSpec(2, 2)
fig = plt.figure('Diagramme', tight_layout=True)

data = np.loadtxt("data/BewegungBeschleunigung.csv", skiprows=1, delimiter=",", unpack=True)
(zeit, strecke, geschwindigkeit, beschleunigung, 
 strecke2, geschwindigkeit2, beschleunigung2) = data


def diagram(y, y2, yachse, titel, pos1, pos2):

    ax = fig.add_subplot(gs[pos1, pos2])
    ax.plot(zeit, y, "go", markersize = 2.5)
    ax.plot(zeit, y2, "ro", markersize = 2.5)
    ax.autoscale(enable = False, axis = "both", tight = None)
    ax.grid(True)
    ax.set_title(titel)

    plt.gcf().canvas.set_window_title("Diagramme")


diagram(strecke, strecke2, "Strecke", "S(t) Diagramm", 0, np.s_[:])
diagram(geschwindigkeit, geschwindigkeit2, "Geschwindigkeit", "V(t) Diagramm", 1, 0)
diagram(beschleunigung, beschleunigung2, "Beschleunigung", "a(t) Diagramm", 1, 1 )
plt.show()

Answer 2

Вы можете поместить условие if в вашу функцию.Вы можете сделать значение по умолчанию pos2 как None.Поэтому, если вы хотите :, вы не передадите никакого значения при вызове diagram.Затем вы проверяете внутри функции, было ли передано какое-либо значение для pos2.Если нет, то вы используете :, иначе вы используете переданное значение.

Ниже приведена измененная функция.У меня нет ваших данных, но это сработает.

---

def diagram(y, y2, yachse, titel, pos1, pos2=None):
    if not pos2:
        ax = fig.add_subplot(gs[pos1, :])
    else:    
        ax = fig.add_subplot(gs[pos1, pos2])
    ax.plot(zeit, y, "go", markersize = 2.5)
    ax.plot(zeit, y2, "ro", markersize = 2.5)
    ax.autoscale(enable = False, axis = "both", tight = None)
    ax.grid(True)
    ax.set_title(titel)

    plt.gcf().canvas.set_window_title("Diagramme")


diagram(strecke, strecke2, "Strecke in Metern/s", "S(t) Diagramm", 0) # <-- No pos2
diagram(geschwindigkeit, geschwindigkeit2, "Geschwindigkeit in Metern/s", "V(t) Diagramm", 1, 0)
diagram(beschleunigung, beschleunigung2, "Beschleunigung in Metern/s^2", "a(t) Diagramm", 1, 1 )
plt.show()