NUMPY034 题解：获取某月所有日期

题目

获取 2016 年 7 月的所有日期。

解题思路

使用 np.arange() 创建日期数组，起止月份作为日期类型。

代码

import numpy as np
Z = np.arange('2016-07', '2016-08', dtype='datetime64[D]')
print(Z)

代码详解

1. np.arange('2016-07', '2016-08') —— 从7月1日到8月1日
2. dtype='datetime64[D]' —— 指定精度为"天"

输出

['2016-07-01' '2016-07-02' '2016-07-03' ... '2016-07-31']

日期精度

# 精度为月
Z = np.arange('2024-01', '2024-03', dtype='datetime64[M]')
# ['2024-01' '2024-02']

# 精度为小时
Z = np.arange('2024-01-01', '2024-01-02', dtype='datetime64[h]')
# ['2024-01-01T00' '2024-01-01T01' ... '2024-01-01T23']

常用日期运算

# 获取月初和月末
first_day = np.datetime64('2024-03', 'D')
last_day = np.datetime64('2024-04', 'D') - np.timedelta64(1, 'D')

# 计算天数
days = (np.datetime64('2024-03-15') - np.datetime64('2024-03-01')).astype(int) + 1

核心知识点

1. np.datetime64 —— 日期时间类型
2. dtype='datetime64[D]' —— 精度设置
3. 日期数组可直接遍历

NumPy034 题解

题目分析

本题要求找出数组中大于k的元素索引。

参考代码

import numpy as np
Z = eval(input())
k = int(input())
print(np.where(Z > k)[0])