INNER CODE UNIT · Python
_simplify_datetimes_in_array
aeturrell/skimpy · src/skimpy/__init__.py:172
def _simplify_datetimes_in_array(rows: np.ndarray) -> np.ndarray:
"""Simplifies 2001/01/01 00:00:00 to 2001/01/01.
Args:
rows (np.ndarray): contain summary info, including datetimes
Returns:
np.ndarray: rows with any all zero hours/min/sec stripped out
"""
timestamp_positions = [
[[idx, i] for i, j in enumerate(item) if isinstance(j, pd.Timestamp)]
for idx, item in enumerate(rows)
]
timestamp_pos_list = list(chain.from_iterable(timestamp_positions))
timestamp_pos_tuples = [tuple(entry) for entry in timestamp_pos_list]
for entry in timestamp_pos_tuples:
hour, min, sec = rows[entry].hour, rows[entry].minute, rows[entry].second
if hour == min == sec == 0: