Treadmill_Paper/step_frequency.py at main · leesuyee/Treadmill_Paper · GitHub

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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import numpy as np

def step_freq(stance_start, time):
    cur_time=time
    step_frequency=[]
#    fig_num=fig_num+1
    for leg in range(0,len(stance_start)):

        # get times of when stance starts
        stance_time=cur_time[stance_start[leg]]

        # calculate the difference in time
        stance_dt=np.diff(stance_time)

        # compute step frequency
        freq=1/stance_dt

        step_frequency.append(freq)

#     # plot step frequency across time
#     # r1
#     plt1=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[0][1::]], step_frequency[0], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('r1 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r1 step frequency'+fig_type
# #     plt1.savefig(fig_name, dpi=dpi_value)

#     # r2
#     plt2=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[1][1::]], step_frequency[1], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('r2 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r2 step frequency'+fig_type
# #     plt2.savefig(fig_name, dpi=dpi_value)

#     # r3
#     plt3=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[2][1::]], step_frequency[2], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('r3 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r3 step frequency'+fig_type
# #     plt3.savefig(fig_name, dpi=dpi_value)

#     # plot step frequency across time
#     # l1
#     plt4=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[3][1::]], step_frequency[3], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('l1 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r1 step frequency'+fig_type
# #     plt1.savefig(fig_name, dpi=dpi_value)

#     # l2
#     plt5=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[4][1::]], step_frequency[4], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('l2 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r2 step frequency'+fig_type
# #     plt2.savefig(fig_name, dpi=dpi_value)

#     # l3
#     plt6=plt.figure(fig_num, figsize=[10,5])
#     plt.plot(cur_time[stance_start[5][1::]], step_frequency[5], color='black', linewidth=1.25)
#     plt.xlabel('time (seconds)', fontsize=18)
#     plt.ylabel('step frequency (1/s)', fontsize=18)
#     plt.title('l3 step frequency', fontsize=18)
#     fig_num=fig_num+1
# #     fig_name= save_dir+'r3 step frequency'+fig_type
# #     plt3.savefig(fig_name, dpi=dpi_value)


    return(step_frequency)