在熊猫中按升序和降序有条件地对单个列进行排序

fjm 发表于 Dev

我有98,000个美国家庭街道地址，我需要按照“步行”的顺序进行排序，即按照您要走的顺序列出，沿着街道的一侧，然后过马路然后向后走。

import pandas as pd
df = pd.read_excel('c:pdsort.xlsx')

# add boolean column for even or odd on number column
is_even = df.loc[:,'number'] % 2 == 0
df.loc[:, 'even'] = is_even

# group and then sort by number
df.groupby(['town','street','even']).apply(lambda x: x.sort_values('number'))

# sort odd numbers ascending and even numbers descending

所需的df结果，将升序排序为奇数街道编号，然后切换为降序排序为偶数。[对不起，第一个stackoverflow问题，尚不具备复制Jupyter笔记本图像的资格]

4列：数字，街道，镇，偶数

列'数字'的期望结果：1231 1233 1235 1237 1239 1238 1236 1234 1232 1230

使用numpy.lexsort，您可以定义要排序的序列序列。来自@smj的数据。

建立

import pandas as pd
import numpy as np

number_list = list(range(1, 11))

df = pd.DataFrame({'town': sorted(['Springfield', 'Shelbyville'] * 10),
                   'street': sorted(['Evergreen Terrace', 'Main Street'] * 10),
                   'number': number_list + number_list})

解

订购时要小心。np.lexsort从序列的最后一个元素开始工作；例如，s1具有最高的排序优先级，s4最低的排序。

s1 = df['town']
s2 = df['street']
s3 = ~df['number']%2                            # i.e. "is odd"
s4 = np.where(s3, -df['number'], df['number'])  # i.e. "negate if odd"

res = df.iloc[np.lexsort((s4, s3, s2, s1))]

结果

print(res)

           town             street  number
0   Shelbyville  Evergreen Terrace       1
2   Shelbyville  Evergreen Terrace       3
4   Shelbyville  Evergreen Terrace       5
6   Shelbyville  Evergreen Terrace       7
8   Shelbyville  Evergreen Terrace       9
9   Shelbyville  Evergreen Terrace      10
7   Shelbyville  Evergreen Terrace       8
5   Shelbyville  Evergreen Terrace       6
3   Shelbyville  Evergreen Terrace       4
1   Shelbyville  Evergreen Terrace       2
10  Springfield        Main Street       1
12  Springfield        Main Street       3
14  Springfield        Main Street       5
16  Springfield        Main Street       7
18  Springfield        Main Street       9
19  Springfield        Main Street      10
17  Springfield        Main Street       8
15  Springfield        Main Street       6
13  Springfield        Main Street       4
11  Springfield        Main Street       2

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。