Python Pandas操作df并创建带有样式和链接的excel

Question

问题 1:对相似行进行分组

我有一个如下所示的数据框:

       TTL1    TTL2    TTL3
0      val1    val2    val3
1      val1    val2    val3
2      val1    val2    val3
3      val1    val2    val3
134    val1    val2    val3
135    val1    val2    val3
136    val1    val2    val3
205    val1    val2    val3

我想检查索引号是否成功(例如 1、2、3 或 56、57、58 等)，如果是，则添加另一列以表明它们在同一组中，因此输出应该是:

       TTL1    TTL2    TTL3   GROUP_TTL
0      val1    val2    val3      0
1      val1    val2    val3      0
2      val1    val2    val3      0
3      val1    val2    val3      0
134    val1    val2    val3      1
135    val1    val2    val3      1
136    val1    val2    val3      1
205    val1    val2    val3      2

Q.2:操作 EXCEL 文件:

现在我不确定可以在 pandas 和 python 中完成的部分:

我想给 excel 中的行着色，这样每组行的颜色都相同。所以第 0 组是绿色，第 1 组是黄色等。

而且我还想在 excel 工作表中添加到计算机上其他一些 excel 工作表的链接。

pandas 和 python 有可能吗？

谢谢!

Answer 1

您可以先使用 convert to_series index 函数 diff (尚未为 index 实现)然后比较值是否不是 1。获取boolean Series 可以通过cumsum 累加和最后减去 1:

df['GROUP_TTL'] = (df.index.to_series().diff() != 1).cumsum().sub(1)

print (df)
     TTL1  TTL2  TTL3  GROUP_TTL
0    val1  val2  val3          0
1    val1  val2  val3          0
2    val1  val2  val3          0
3    val1  val2  val3          0
134  val1  val2  val3          1
135  val1  val2  val3          1
136  val1  val2  val3          1
205  val1  val2  val3          2

着色:

#http://xlsxwriter.readthedocs.io/example_pandas_conditional.html
len_index = len(df.index) + 1
writer = pd.ExcelWriter('file.xlsx', engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', index=None)
workbook  = writer.book
worksheet = writer.sheets['Sheet1']
worksheet.conditional_format('D2:D'+str(len_index), {'type': '3_color_scale'})
writer.save()