我有一個數據框,其中包含一個名為“id”的列,其中每個 id 的長度為 9 個字符,我正在嘗試添加一個列“Rating”,我根據第 6 行將每一行評為 A、AA 或 AAA,第 7 個和第 8 個字符是“A00”、“AA0”或“000”。到目前為止,我有以下代碼:id = df['id']conditions = [(id.str.get(5) == 'A00'), (id.str.get(5) == 'AA0'), (id.str.get(5) == '000')]values = ['A', 'AA', 'AAA']df['Rating'] = np.select(conditions, values)df['Rating'] = df['Rating'].astype('category')但我知道條件行是錯誤的,因為 column.str.get(n) 只檢查字符串的第 (n-1) 個字符,我需要長度為 3 的子字符串,而不僅僅是一個字符。有誰知道我可以使用哪個命令?提前致謝!
通過引用字符串位置檢查數據框列中的子字符串
慕田峪9158850
2023-03-16 09:34:11