Python - 如何将列映射到字母序列？

Question

这个问题已经有答案了:

How to create an infinite iterator to generate an incrementing alphabet pattern? (2 个回答)

已关闭 3 年前。

我在数据框中有一列，我想将其映射到一系列字母。

假设该列中有以下 50 个值:

import pandas as pd
import numpy as np

data = pd.DataFrame(np.arange(50))

现在我想创建一个新列，其中数字0映射到A，1映射到B等。我遵循this question中的方法，但是当映射到达字母 Z 时，它会转到一些非字母字符。这是映射代码:

data['new column'] = [chr(ord('a') + x).upper() for x in data.index]    

观察数字 26 及之后的数字被映射到非字母字符。我们如何创建一个映射函数，以便映射将数字 26 重新映射到 AA、27 到 AB、28 到 AC 等？

Answer 1

import pandas as pd
import numpy as np

data = pd.DataFrame(np.arange(50))

data["new column"] = [
    chr(ord('a') + x//26 - 1).upper() + chr(ord('a') + x%26).upper() if x >= 26
    else chr(ord('a') + x).upper()
    for x in data.index
]

祝你好运