File size: 1,157 Bytes
97869d5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import random
from datetime import datetime, timedelta
def generate_data(start_time: datetime, end_time: datetime, window_size: int):
"""
Generate random traffic volume data for a given time range and window size.
"""
traffic_data = []
current_time = start_time
while current_time < end_time:
# Generate random traffic volume data for the current window
traffic_volume = random.randint(0, 100)
traffic_data.append({
'time': current_time.strftime("%Y-%m-%d %H:%M:%S"),
'volume': traffic_volume
})
current_time += timedelta(minutes=window_size)
return traffic_data
if __name__ == "__main__":
start_time = datetime(2023, 10, 1, 8, 0, 0) # October 1, 2023, 08:00:00
end_time = datetime(2023, 10, 1, 10, 0, 0) # October 1, 2023, 10:00:00
window_size = 15 # 15-minute window
traffic_data = generate_data(start_time, end_time, window_size)
with open('data/traffic_data.csv', 'w') as f:
f.write("time,volume\n")
for data in traffic_data:
f.write(f"{data['time']},{data['volume']}\n")
|