熊猫数据帧范围索引

Question

我创建了一个Pandas DataFrame。我需要为与帧对应的数据帧创建一个范围索引 - RangeIndex(start=0, stop=x, step=y) - x和y与我的数据帧相关的地方。

我还没有看到如何做到这一点的例子 - 是否有特定于此的方法或语法？

谢谢

Answer 1

似乎你需要RangeIndex构造函数：

df = pd.DataFrame({'A' : range(1, 21)})
print (df)
     A
0    1
1    2
2    3
3    4
4    5
5    6
6    7
7    8
8    9
9   10
10  11
11  12
12  13
13  14
14  15
15  16
16  17
17  18
18  19
19  20

print (df.index)
RangeIndex(start=0, stop=20, step=1)

---

df.index = pd.RangeIndex(start=0, stop=99, step=5)
print (df)
     A
0    1
5    2
10   3
15   4
20   5
25   6
30   7
35   8
40   9
45  10
50  11
55  12
60  13
65  14
70  15
75  16
80  17
85  18
90  19
95  20
print (df.index)
RangeIndex(start=0, stop=99, step=5)

更动态的解决方案：

step = 10
df.index = pd.RangeIndex(start=0, stop=len(df.index) * step - 1, step=step)
print (df)
      A
0     1
10    2
20    3
30    4
40    5
50    6
60    7
70    8
80    9
90   10
100  11
110  12
120  13
130  14
140  15
150  16
160  17
170  18
180  19
190  20
print (df.index)
RangeIndex(start=0, stop=199, step=10)

编辑：

正如@ZakS注释中指出的那样，更好的是仅DataFrame构造函数一起使用：

df = pd.DataFrame({'A' : range(1, 21)}, index=pd.RangeIndex(start=0, stop=99, step=5))
print (df)
0    1
5    2
10   3
15   4
20   5
25   6
30   7
35   8
40   9
45  10
50  11
55  12
60  13
65  14
70  15
75  16
80  17
85  18
90  19
95  20