Python 3x。熊猫从csv获取的特定index_col

Question

我正在尝试在代码中添加打印内容，其中将显示用于特定index_col的行数。

[请让我们看下面的示例。

import pandas as pd
def a2c():
data = pd.read_csv("Claims.csv", sep=";", index_col="A2C")
pd.set_option("display.max_rows", None)
while True:
    try:
        vstup=input("Zadejte A2C: ")
        print(data.loc[vstup])
    except KeyError:
        print("Špatná hodnota 1")
        break

def qn():
    data = pd.read_csv("Claims.csv", sep=";", index_col="QN")
    pd.set_option("display.max_rows", None)
    while True:
        try:
           # print( data.dtypes)
            vstup=(input("Zadejte QN: "))
            print(data.loc[vstup])
        except KeyError:
            print("Špatná hodnota 1")
            break

while True:
    print("1)  QN \t\t = Číslo 13tkového hlášení\n2)  Druh \t = CO/CW/DD/II/PC\n3)  NR \t\t = Číslo NR hlášení\n4)  Zákazník \t = Název Zákazníka(Vše velkýmy písmeny)\n5)  Dodavatel\t = Celý název dodavatele\n6)  FF \t\t = BDY_FF1 / BDY_FF2 / BDY_FF3  -> Lze použít pro přehled nezpracovaných reklamací při volbě 0\n7)  A2C \t = Čislo A2C komponentu\n8)  Dod \t = Číslo dodavatele\n9)  UIDA \t = Číslo UIDY\n10) L \t\t = Číslo první Liability\n11) Cycle \t = Datum zastavení Cycle Time ")
    vstup=input("\nZadejte co hledáte: ")
    if vstup == "6":
        ff()
    elif vstup == "Zalozeno":
        print("Zadejte datum založení reklamace ve formě DD/MM/YY")
        zalozeno()
    elif vstup == "1":
        qn()

如果我尝试添加到那里：

vstup=input("Zadejte A2C: ")
print(data.loc[vstup])
print(len(data)

例如，它计算了cvs文件中的所有行，而不是索引A2C。

我可以用来查看特定索引的行数吗？

谢谢您的帮助。

Answer 1

只需将您的代码更改为此

vstup=input("Zadejte A2C: ")
print(sum(df.index == vstup))

它将为您提供与给定索引对应的行数