在 pandas 中使用 str[:] 時出現 KeyError

Question

我有一個帶有行 ID 的表：

ID	水果	地方
61725RN	蘋果	倫敦
12381YH	香蕉	多倫多

我需要列出一個只有一部分 ID 的列表：['61725', '12381']

我在下面編寫代碼並收到“KeyError”。 但是，列名在我的代碼中是粗體的，所以它是一個索引。

nums = df['ID'].str[:5]

Answer 1

您必須在切片之前將目標轉換為字符串，編輯：並且您必須進行循環：

    nums=[]
    for i in range(len(df)):
        nums.append(str(df['ID'][i])[:5])

抱歉，我沒有檢查 dataframe 問題。 現在工作。