python - 创建一个新列作为 Pandas DataFrame 的计数

Question

我有一个 Pandas DataFrame。如何创建一个类似于 Pandas DataFrame 计数的新列，因为我已经将索引设为 Datatime。

例如，以下代码可在您的本地 PC 上重现:

import datetime
import numpy

dates = [
    datetime.date(2019, 1, 13),
    datetime.date(2020, 5, 11),
    datetime.date(2018, 7, 24),
    datetime.date(2019, 3, 23),
    datetime.date(2020, 2, 16)
]

data = {
    "a": [13.3,12.3,np.nan,10.3,np.nan],
    "b": [1,0,0,1,1],
    "c": ["no","yes","no","","yes"]
}

pd.DataFrame(index=dates,data=data)

现在，我想添加一个新列作为计数。类似于 1,2,3,4,5 直到数据结束

Answer 1

df['count'] = range(1, len(df) + 1)

len(df)返回 DataFrame 中的行数，因此您可以调用内置 range从 1 创建范围的函数到 DataFrame 中的行数，然后将其分配给新列。为列分配范围时，它会自动转换为 Pandas 系列。