9，Python数分之Pandas训练，力扣，1831. 每天的最大交易

最新推荐文章于 2024-09-29 00:25:06 发布

凡梦_leo

最新推荐文章于 2024-09-29 00:25:06 发布

阅读量947

点赞数 29

分类专栏：数分之Pandas实战训练文章标签： python pandas leetcode

本文链接：https://blog.csdn.net/qq_55006020/article/details/142443035

版权

数分之Pandas实战训练专栏收录该内容

32 篇文章 0 订阅

订阅专栏

学习：知识的初次邂逅
复习：知识的温故知新
练习：知识的实践应用

一，原题力扣链接

. - 力扣（LeetCode）

二，题干

表：Transactions

+----------------+----------+
| Column Name    | Type     |
+----------------+----------+
| transaction_id | int      |
| day            | datetime |
| amount         | int      |
+----------------+----------+
transaction_id 是该表具有唯一值的列。
每行包括了该次交易的信息。

编写一个解决方案，报告每天交易金额 amount 最大的交易 ID 。如果一天中有多个这样的交易，返回这些交易的 ID 。

返回结果根据 transaction_id 升序排列。

返回格式如下示例所示：

示例 1：

输入：
Transactions table:
+----------------+--------------------+--------+
| transaction_id | day                | amount |
+----------------+--------------------+--------+
| 8              | 2021-4-3 15:57:28  | 57     |
| 9              | 2021-4-28 08:47:25 | 21     |
| 1              | 2021-4-29 13:28:30 | 58     |
| 5              | 2021-4-28 16:39:59 | 40     |
| 6              | 2021-4-29 23:39:28 | 58     |
+----------------+--------------------+--------+
输出：
+----------------+
| transaction_id |
+----------------+
| 1              |
| 5              |
| 6              |
| 8              |
+----------------+
解释：
"2021-4-3"  --> 有一个 id 是 8 的交易，因此，把它加入结果表。 
"2021-4-28" --> 有两个交易，id 是 5 和 9 ，交易 5 的金额是 40 ，而交易 9 的数量是 21 。只需要将交易 5 加入结果表，因为它是当天金额最大的交易。
"2021-4-29" --> 有两个交易，id 是 1 和 6 ，这两个交易的金额都是 58 ，因此需要把它们都写入结果表。
最后，把交易 id 按照升序排列。

进阶：你可以不使用 MAX() 函数解决这道题目吗?

回答: 本来就没用max() 函数解决这个问题用开窗排序

三，建表语句

import pandas as pd
data = [[8, '2021-4-3 15:57:28', 57], [9, '2021-4-28 08:47:25', 21], [1, '2021-4-29 13:28:30', 58], [5, '2021-4-28 16:39:59', 40], [6, '2021-4-29 23:39:28', 58]]
transactions = pd.DataFrame(data, columns=['transaction_id', 'day', 'amount']).astype({'transaction_id':'Int64', 'day':'datetime64[ns]', 'amount':'Int64'})

四，分析

题解:

表:交易类型表

字段:交易id,交易日期,交易金额

求:每日最大的交易允许并列按照要求排序

第一步:去day的列的年月日

transactions['date_only'] = transactions['day'].astype(str).str.split().str[0]  #截取年月日

第二步:开一个排序列允许并列所以用rank 以年月日分组以金额排序

transactions['rn'] = transactions.groupby('date_only')['amount'].rank(method='min',ascending=False)  #开一个排序列 rn
transactions

第三步: 过滤rn ==1 的值以及映射指定的列并按照要求排序

五，Pandas解答

import pandas as pd

def find_maximum_transaction(transactions: pd.DataFrame) -> pd.DataFrame:
    transactions['date_only'] = transactions['day'].astype(str).str.split().str[0]
    transactions['rn'] = transactions.groupby('date_only')['amount'].rank(method='min',ascending=False)
    df =transactions[transactions['rn'] ==1]
    df2 =df.sort_values('transaction_id',ascending=True)
    df3 = df2['transaction_id']
    df4 =df3.to_frame()
    return df4