# Write your MySQL query statement below
601. 体育馆的人流量
表：Stadium
+---------------+---------+ | Column Name | Type | +---------------+---------+ | id | int | | visit_date | date | | people | int | +---------------+---------+ visit_date 是表的主键 每日人流量信息被记录在这三列信息中：序号 (id)、日期 (visit_date)、 人流量 (people) 每天只有一行记录，日期随着 id 的增加而增加
编写一个 SQL 查询以找出每行的人数大于或等于 100 且 id 连续的三行或更多行记录。
返回按 visit_date 升序排列 的结果表。
查询结果格式如下所示。
示例 1:
输入：
Stadium 表:
+------+------------+-----------+
| id | visit_date | people |
+------+------------+-----------+
| 1 | 2017-01-01 | 10 |
| 2 | 2017-01-02 | 109 |
| 3 | 2017-01-03 | 150 |
| 4 | 2017-01-04 | 99 |
| 5 | 2017-01-05 | 145 |
| 6 | 2017-01-06 | 1455 |
| 7 | 2017-01-07 | 199 |
| 8 | 2017-01-09 | 188 |
+------+------------+-----------+
输出：
+------+------------+-----------+
| id | visit_date | people |
+------+------------+-----------+
| 5 | 2017-01-05 | 145 |
| 6 | 2017-01-06 | 1455 |
| 7 | 2017-01-07 | 199 |
| 8 | 2017-01-09 | 188 |
+------+------------+-----------+
解释：
id 为 5、6、7、8 的四行 id 连续，并且每行都有 >= 100 的人数记录。
请注意，即使第 7 行和第 8 行的 visit_date 不是连续的，输出也应当包含第 8 行，因为我们只需要考虑 id 连续的记录。
不输出 id 为 2 和 3 的行，因为至少需要三条 id 连续的记录。
原站题解
pythondata 解法, 执行用时: 355 ms, 内存消耗: 67.1 MB, 提交时间: 2024-05-27 10:31:17
import pandas as pd
def human_traffic(stadium: pd.DataFrame) -> pd.DataFrame:
# 使用条件过滤，仅选择满足人数要求的记录
filtered_stadium = stadium[stadium['people'] >= 100]
# 使用 rank 方法计算排名差异
filtered_stadium['rnk_diff'] = filtered_stadium['id'] - filtered_stadium['id'].rank(method='first')
# 使用 groupby 和 having 筛选出排名差异满足条件的组
valid_rnk_diff = filtered_stadium.groupby('rnk_diff').filter(lambda group: len(group) >= 3)['rnk_diff'].unique()
# 选择符合排名差异条件的记录
result = filtered_stadium[filtered_stadium['rnk_diff'].isin(valid_rnk_diff)][['id', 'visit_date', 'people']]
# 按照访问日期排序
result = result.sort_values(by='visit_date')
return result
mysql 解法, 执行用时: 384 ms, 内存消耗: 0 B, 提交时间: 2024-05-27 10:30:50
# Write your MySQL query statement below
with t1 as(
select *,id - row_number() over(order by id) as rk
from stadium
where people >= 100
)
select id,visit_date,people
from t1
where rk in(
select rk
from t1
group by rk
having count(rk) >= 3
)
mysql 解法, 执行用时: 557 ms, 内存消耗: N/A, 提交时间: 2018-08-22 16:13:32
# Write your MySQL query statement below select distinct s1.* from stadium s1, stadium s2, stadium s3 where s1.people >= 100 and s2.people >= 100 and s3.people >= 100 and ( (s1.id - s2.id = 1 and s2.id - s3.id = 1) or (s2.id - s1.id = 1 and s1.id - s3.id = 1) or (s3.id - s2.id = 1 and s2.id - s1.id = 1) ) order by s1.id;