表格与特定条件的组合

时间:2018-10-18 01:35:01

标签: python combinations itertools

我有一个表,其中Day是列标题,而温度是行。该表列出了每天具有给定最低温度的次数。在下表中,星期一的最低温度为1度的4倍,星期二的最低温度为1度的0倍,星期三的最低温度为1度的3倍,依此类推。

     **Mon**       **Tue**      **Wed**     
1      4              0            3
2      0              1            0
3      1              4            2
4      2              0            1
5      0              1            3    
6      7              3            2
7      0              0            1
8      2              1            4

我想生成温度和温度的所有可能组合(a,b,c) 表示等距值(例如(a-b) = 3(b-c) = 0)的天数。

星期一,最低温度为1度= 4 星期二,最低温度为2度= 1 差异= 3

星期二,最低温度为2度= 1 星期三,最低温度4度= 1 差异= 0

将输出值作为行标签:(1,2,4)(1,2,7)(6,3,8)

我应该如何在整个表中获得等距值的相似组合?

1 个答案:

答案 0 :(得分:0)

不确定我是否清楚要求。但是请检查这是否是您需要的:

dfb['Group'] = groups[np.searchsorted(dfb['TimeStamp'].values, vals) // 2]

print(dfb)

   TimeStamp Group
0          1     A
1          2     A
2          3     B
3          4     B
4          5     C
5          6     C
6          7     D
7          8     D