读取gzmt.csv文件，计算均值及概率

问题：

读取gzmt.csv文件所有数据，选取收盘价格（倒数第二列），计算20天均值，权重取成交量（选做：时间权重为半衰期为15天）；将该均值修剪为超过600的都设置为1000，并打印出该均值超过550或者低于400的日期是周几的概率，比如周一的概率，周二的概率等。

实现：

import numpy as np
import datetime

close, num = np.loadtxt('gzmt.csv', delimiter=',', usecols=(-2, -1), unpack=True)
price_avg = [np.average(close[i:20+i], weights=num[i:20+i]) for i in range(len(close)-20)]
print(price_avg)

price_avg = [i if i <= 600 else 1000 for i in price_avg]
print(price_avg)

price_avg = tuple(price_avg)
dates = np.loadtxt('gzmt.csv', dtype=np.str, delimiter=',', usecols=(0))
date = [datetime.datetime.strptime(i, '%Y/%m/%d') for i in dates]
print(date)
date = [i.date().weekday() for i in date]
date_dict = dict(zip(price_avg, tuple(date[:220])))
d = {k:w for k,w in date_dict.items() if k < 400 or k > 550}
Mon = list(d.values()).count(0)/len(d.values())
print('星期一：', format(Mon, '.2%'))
Tues = list(d.values()).count(1)/len(d.values())
print('星期二：', format(Tues,'.2%'))
Wen = list(d.values()).count(2)/len(d.values())
print('星期三：', format(Wen, '.2%'))
Thurs = list(d.values()).count(3)/len(d.values())
print('星期四：', format(Thurs, '.2%'))
Fri = list(d.values()).count(4)/len(d.values())
print('星期五：',format(Fri,'.2%'))
Satur = list(d.values()).count(5)/len(d.values())
print('星期六：', format(Satur,'.2%'))
Sun = list(d.values()).count(6)/len(d.values())
print('星期日：', format(Sun, '.2%'))

 输出结果：

[341.94166218833112, 342.23298251142694, 341.76667746967502, 341.85881034007457, 341.432513273224, 341.56123946251921, 341.61341553032884, 341.74514867178249, 341.77862536719448, 342.00088575238277, 343.76861119378776, 344.24404843818115, 344.62225399817237, 344.87847481075642, 345.09602554564202, 345.37701234743838, 345.5892682794381, 345.95190335704257, 346.69984621828183, 347.04063152823204, 347.45874208114776, 348.90082519336204, 349.6183553956941, 350.54123494099201, 351.92983426128825, 353.43426138322178, 354.26309509706147, 355.48803567065551, 356.50105396149746, 358.10026196103792, 359.79826104134986, 362.6010843906127, 365.13789984402052, 366.63440294348453, 367.71992642615567, 368.81083231596716, 370.03080347599257, 371.11462477934521, 372.22574756951246, 372.97545685178335, 374.26329664525468, 375.69951498884177, 376.4340041671739, 377.14830096910282, 378.08228334059618, 378.83718030241226, 380.13010568016705, 380.85422245889009, 381.64103208816294, 383.23216496945565, 384.19885687771097, 385.6785385665263, 386.09545767436344, 386.61596649270911, 389.34216542825578, 392.06402708504305, 393.85754496714873, 395.45443790752842, 396.39299177090504, 397.2930530860807, 398.47316599560509, 399.49580472257401, 399.86189312853895, 400.61447223927001, 401.90539583018239, 402.65747042487124, 403.41975227169291, 404.25716943085587, 405.97817697148901, 406.97050971837933, 407.85287918041905, 409.12189305480319, 411.35272388306055, 414.99839023725974, 416.92001525388656, 418.53101898034919, 419.75695837034419, 421.78124899839673, 423.54197692029135, 424.44226381795255, 425.39967231914909, 426.79438728489748, 429.99303718595201, 433.51173779465501, 435.94315250245421, 438.97179272641068, 441.06984949208606, 443.8650826643817, 446.46658030492375, 448.03909077251768, 449.81393142003549, 451.20431516214711, 452.95048514001388, 454.0902499269917, 455.29023434587879, 457.51392053649937, 458.81244560284262, 460.82384175419173, 462.20591539533336, 462.99911863829766, 463.02864757173785, 462.24421881909285, 462.29956222296312, 461.75248389739198, 461.08431085175215, 460.53609486235416, 460.50696845435078, 460.08608794305565, 460.12249212882432, 460.22671793437075, 459.57397516027208, 459.28217537617849, 458.72490578304547, 458.83703104943135, 459.42174008137653, 459.16807020234853, 459.04119998289252, 459.09333579922964, 459.1473006034243, 460.50072566081354, 462.03188260060625, 464.0672480755689, 465.35259090356811, 466.87484366762851, 468.23272132387058, 469.46020585392853, 470.2390737397925, 472.14560293923154, 473.83058969322309, 474.84385035746232, 477.58162745834983, 479.46081161765227, 481.95436906791184, 483.57689637736564, 484.63980590930629, 485.35766788662102, 485.76355916560681, 486.89735891448788, 487.51323145778326, 487.98196802785623, 488.66824017836984, 489.12686856115818, 489.57068888522201, 490.37217754240254, 491.23226674399911, 491.69185515345384, 491.89080875580817, 491.54436433427963, 490.8937036475831, 490.08980612682143, 488.39592554733713, 487.80271660375297, 487.74077094200021, 487.61444750233699, 487.59100372768819, 488.34853615600832, 488.79775570327018, 488.77237903649933, 488.99606903581434, 489.19962071593903, 490.13556308748792, 491.06719966814012, 491.5306910624513, 492.82019875376363, 493.71588288903365, 495.9017009510211, 498.57025245989189, 502.15020291572665, 504.91365004462875, 509.9171762627916, 515.58213716880607, 519.35639338785541, 522.81949174598265, 528.74699578746163, 532.63189828997338, 537.29827319564879, 540.89043659513288, 543.53220218657543, 556.3073844561859, 569.01418927134159, 577.54636276986071, 582.55131429850076, 586.32674940428603, 590.25785751423552, 595.07359097571498, 601.20805213914616, 606.0195190750768, 610.54909312773964, 613.50837171811452, 620.31056355735996, 625.69683247883768, 629.67483534258486, 633.59660988971598, 640.66442657593586, 648.01383970322786, 651.83123178156154, 655.89928250322964, 657.0606746869903, 659.61583269966786, 658.55473519724478, 658.99852153020754, 660.01945301793012, 660.45366904161745, 660.47497105609477, 659.90116735542881, 659.25919622230856, 659.04104972345147, 657.86376378164596, 656.52692832439413, 654.36087821227432, 652.96154460887112, 652.03243812318772, 651.42242099801422, 648.68150018628205, 644.70923570753462, 642.87061813063133, 640.93442770887179, 642.3720801922907, 645.0869567024663, 647.36103350197186]
[341.94166218833112, 342.23298251142694, 341.76667746967502, 341.85881034007457, 341.432513273224, 341.56123946251921, 341.61341553032884, 341.74514867178249, 341.77862536719448, 342.00088575238277, 343.76861119378776, 344.24404843818115, 344.62225399817237, 344.87847481075642, 345.09602554564202, 345.37701234743838, 345.5892682794381, 345.95190335704257, 346.69984621828183, 347.04063152823204, 347.45874208114776, 348.90082519336204, 349.6183553956941, 350.54123494099201, 351.92983426128825, 353.43426138322178, 354.26309509706147, 355.48803567065551, 356.50105396149746, 358.10026196103792, 359.79826104134986, 362.6010843906127, 365.13789984402052, 366.63440294348453, 367.71992642615567, 368.81083231596716, 370.03080347599257, 371.11462477934521, 372.22574756951246, 372.97545685178335, 374.26329664525468, 375.69951498884177, 376.4340041671739, 377.14830096910282, 378.08228334059618, 378.83718030241226, 380.13010568016705, 380.85422245889009, 381.64103208816294, 383.23216496945565, 384.19885687771097, 385.6785385665263, 386.09545767436344, 386.61596649270911, 389.34216542825578, 392.06402708504305, 393.85754496714873, 395.45443790752842, 396.39299177090504, 397.2930530860807, 398.47316599560509, 399.49580472257401, 399.86189312853895, 400.61447223927001, 401.90539583018239, 402.65747042487124, 403.41975227169291, 404.25716943085587, 405.97817697148901, 406.97050971837933, 407.85287918041905, 409.12189305480319, 411.35272388306055, 414.99839023725974, 416.92001525388656, 418.53101898034919, 419.75695837034419, 421.78124899839673, 423.54197692029135, 424.44226381795255, 425.39967231914909, 426.79438728489748, 429.99303718595201, 433.51173779465501, 435.94315250245421, 438.97179272641068, 441.06984949208606, 443.8650826643817, 446.46658030492375, 448.03909077251768, 449.81393142003549, 451.20431516214711, 452.95048514001388, 454.0902499269917, 455.29023434587879, 457.51392053649937, 458.81244560284262, 460.82384175419173, 462.20591539533336, 462.99911863829766, 463.02864757173785, 462.24421881909285, 462.29956222296312, 461.75248389739198, 461.08431085175215, 460.53609486235416, 460.50696845435078, 460.08608794305565, 460.12249212882432, 460.22671793437075, 459.57397516027208, 459.28217537617849, 458.72490578304547, 458.83703104943135, 459.42174008137653, 459.16807020234853, 459.04119998289252, 459.09333579922964, 459.1473006034243, 460.50072566081354, 462.03188260060625, 464.0672480755689, 465.35259090356811, 466.87484366762851, 468.23272132387058, 469.46020585392853, 470.2390737397925, 472.14560293923154, 473.83058969322309, 474.84385035746232, 477.58162745834983, 479.46081161765227, 481.95436906791184, 483.57689637736564, 484.63980590930629, 485.35766788662102, 485.76355916560681, 486.89735891448788, 487.51323145778326, 487.98196802785623, 488.66824017836984, 489.12686856115818, 489.57068888522201, 490.37217754240254, 491.23226674399911, 491.69185515345384, 491.89080875580817, 491.54436433427963, 490.8937036475831, 490.08980612682143, 488.39592554733713, 487.80271660375297, 487.74077094200021, 487.61444750233699, 487.59100372768819, 488.34853615600832, 488.79775570327018, 488.77237903649933, 488.99606903581434, 489.19962071593903, 490.13556308748792, 491.06719966814012, 491.5306910624513, 492.82019875376363, 493.71588288903365, 495.9017009510211, 498.57025245989189, 502.15020291572665, 504.91365004462875, 509.9171762627916, 515.58213716880607, 519.35639338785541, 522.81949174598265, 528.74699578746163, 532.63189828997338, 537.29827319564879, 540.89043659513288, 543.53220218657543, 556.3073844561859, 569.01418927134159, 577.54636276986071, 582.55131429850076, 586.32674940428603, 590.25785751423552, 595.07359097571498, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000, 1000]
[datetime.datetime(2017, 1, 3, 0, 0), datetime.datetime(2017, 1, 4, 0, 0), datetime.datetime(2017, 1, 5, 0, 0), datetime.datetime(2017, 1, 6, 0, 0), datetime.datetime(2017, 1, 9, 0, 0), datetime.datetime(2017, 1, 10, 0, 0), datetime.datetime(2017, 1, 11, 0, 0), datetime.datetime(2017, 1, 12, 0, 0), datetime.datetime(2017, 1, 13, 0, 0), datetime.datetime(2017, 1, 16, 0, 0), datetime.datetime(2017, 1, 17, 0, 0), datetime.datetime(2017, 1, 18, 0, 0), datetime.datetime(2017, 1, 19, 0, 0), datetime.datetime(2017, 1, 20, 0, 0), datetime.datetime(2017, 1, 23, 0, 0), datetime.datetime(2017, 1, 24, 0, 0), datetime.datetime(2017, 1, 25, 0, 0), datetime.datetime(2017, 1, 26, 0, 0), datetime.datetime(2017, 2, 3, 0, 0), datetime.datetime(2017, 2, 6, 0, 0), datetime.datetime(2017, 2, 7, 0, 0), datetime.datetime(2017, 2, 8, 0, 0), datetime.datetime(2017, 2, 9, 0, 0), datetime.datetime(2017, 2, 10, 0, 0), datetime.datetime(2017, 2, 13, 0, 0), datetime.datetime(2017, 2, 14, 0, 0), datetime.datetime(2017, 2, 15, 0, 0), datetime.datetime(2017, 2, 16, 0, 0), datetime.datetime(2017, 2, 17, 0, 0), datetime.datetime(2017, 2, 20, 0, 0), datetime.datetime(2017, 2, 21, 0, 0), datetime.datetime(2017, 2, 22, 0, 0), datetime.datetime(2017, 2, 23, 0, 0), datetime.datetime(2017, 2, 24, 0, 0), datetime.datetime(2017, 2, 27, 0, 0), datetime.datetime(2017, 2, 28, 0, 0), datetime.datetime(2017, 3, 1, 0, 0), datetime.datetime(2017, 3, 2, 0, 0), datetime.datetime(2017, 3, 3, 0, 0), datetime.datetime(2017, 3, 6, 0, 0), datetime.datetime(2017, 3, 7, 0, 0), datetime.datetime(2017, 3, 8, 0, 0), datetime.datetime(2017, 3, 9, 0, 0), datetime.datetime(2017, 3, 10, 0, 0), datetime.datetime(2017, 3, 13, 0, 0), datetime.datetime(2017, 3, 14, 0, 0), datetime.datetime(2017, 3, 15, 0, 0), datetime.datetime(2017, 3, 16, 0, 0), datetime.datetime(2017, 3, 17, 0, 0), datetime.datetime(2017, 3, 20, 0, 0), datetime.datetime(2017, 3, 21, 0, 0), datetime.datetime(2017, 3, 22, 0, 0), datetime.datetime(2017, 3, 23, 0, 0), datetime.datetime(2017, 3, 24, 0, 0), datetime.datetime(2017, 3, 27, 0, 0), datetime.datetime(2017, 3, 28, 0, 0), datetime.datetime(2017, 3, 29, 0, 0), datetime.datetime(2017, 3, 30, 0, 0), datetime.datetime(2017, 3, 31, 0, 0), datetime.datetime(2017, 4, 5, 0, 0), datetime.datetime(2017, 4, 6, 0, 0), datetime.datetime(2017, 4, 7, 0, 0), datetime.datetime(2017, 4, 10, 0, 0), datetime.datetime(2017, 4, 11, 0, 0), datetime.datetime(2017, 4, 12, 0, 0), datetime.datetime(2017, 4, 13, 0, 0), datetime.datetime(2017, 4, 14, 0, 0), datetime.datetime(2017, 4, 17, 0, 0), datetime.datetime(2017, 4, 18, 0, 0), datetime.datetime(2017, 4, 19, 0, 0), datetime.datetime(2017, 4, 20, 0, 0), datetime.datetime(2017, 4, 21, 0, 0), datetime.datetime(2017, 4, 24, 0, 0), datetime.datetime(2017, 4, 25, 0, 0), datetime.datetime(2017, 4, 26, 0, 0), datetime.datetime(2017, 4, 27, 0, 0), datetime.datetime(2017, 4, 28, 0, 0), datetime.datetime(2017, 5, 2, 0, 0), datetime.datetime(2017, 5, 3, 0, 0), datetime.datetime(2017, 5, 4, 0, 0), datetime.datetime(2017, 5, 5, 0, 0), datetime.datetime(2017, 5, 8, 0, 0), datetime.datetime(2017, 5, 9, 0, 0), datetime.datetime(2017, 5, 10, 0, 0), datetime.datetime(2017, 5, 11, 0, 0), datetime.datetime(2017, 5, 12, 0, 0), datetime.datetime(2017, 5, 15, 0, 0), datetime.datetime(2017, 5, 16, 0, 0), datetime.datetime(2017, 5, 17, 0, 0), datetime.datetime(2017, 5, 18, 0, 0), datetime.datetime(2017, 5, 19, 0, 0), datetime.datetime(2017, 5, 22, 0, 0), datetime.datetime(2017, 5, 23, 0, 0), datetime.datetime(2017, 5, 24, 0, 0), datetime.datetime(2017, 5, 25, 0, 0), datetime.datetime(2017, 5, 26, 0, 0), datetime.datetime(2017, 5, 31, 0, 0), datetime.datetime(2017, 6, 1, 0, 0), datetime.datetime(2017, 6, 2, 0, 0), datetime.datetime(2017, 6, 5, 0, 0), datetime.datetime(2017, 6, 6, 0, 0), datetime.datetime(2017, 6, 7, 0, 0), datetime.datetime(2017, 6, 8, 0, 0), datetime.datetime(2017, 6, 9, 0, 0), datetime.datetime(2017, 6, 12, 0, 0), datetime.datetime(2017, 6, 13, 0, 0), datetime.datetime(2017, 6, 14, 0, 0), datetime.datetime(2017, 6, 15, 0, 0), datetime.datetime(2017, 6, 16, 0, 0), datetime.datetime(2017, 6, 19, 0, 0), datetime.datetime(2017, 6, 20, 0, 0), datetime.datetime(2017, 6, 21, 0, 0), datetime.datetime(2017, 6, 22, 0, 0), datetime.datetime(2017, 6, 23, 0, 0), datetime.datetime(2017, 6, 26, 0, 0), datetime.datetime(2017, 6, 27, 0, 0), datetime.datetime(2017, 6, 28, 0, 0), datetime.datetime(2017, 6, 29, 0, 0), datetime.datetime(2017, 6, 30, 0, 0), datetime.datetime(2017, 7, 3, 0, 0), datetime.datetime(2017, 7, 4, 0, 0), datetime.datetime(2017, 7, 5, 0, 0), datetime.datetime(2017, 7, 6, 0, 0), datetime.datetime(2017, 7, 7, 0, 0), datetime.datetime(2017, 7, 10, 0, 0), datetime.datetime(2017, 7, 11, 0, 0), datetime.datetime(2017, 7, 12, 0, 0), datetime.datetime(2017, 7, 13, 0, 0), datetime.datetime(2017, 7, 14, 0, 0), datetime.datetime(2017, 7, 17, 0, 0), datetime.datetime(2017, 7, 18, 0, 0), datetime.datetime(2017, 7, 19, 0, 0), datetime.datetime(2017, 7, 20, 0, 0), datetime.datetime(2017, 7, 21, 0, 0), datetime.datetime(2017, 7, 24, 0, 0), datetime.datetime(2017, 7, 25, 0, 0), datetime.datetime(2017, 7, 26, 0, 0), datetime.datetime(2017, 7, 27, 0, 0), datetime.datetime(2017, 7, 28, 0, 0), datetime.datetime(2017, 7, 31, 0, 0), datetime.datetime(2017, 8, 1, 0, 0), datetime.datetime(2017, 8, 2, 0, 0), datetime.datetime(2017, 8, 3, 0, 0), datetime.datetime(2017, 8, 4, 0, 0), datetime.datetime(2017, 8, 7, 0, 0), datetime.datetime(2017, 8, 8, 0, 0), datetime.datetime(2017, 8, 9, 0, 0), datetime.datetime(2017, 8, 10, 0, 0), datetime.datetime(2017, 8, 11, 0, 0), datetime.datetime(2017, 8, 14, 0, 0), datetime.datetime(2017, 8, 15, 0, 0), datetime.datetime(2017, 8, 16, 0, 0), datetime.datetime(2017, 8, 17, 0, 0), datetime.datetime(2017, 8, 18, 0, 0), datetime.datetime(2017, 8, 21, 0, 0), datetime.datetime(2017, 8, 22, 0, 0), datetime.datetime(2017, 8, 23, 0, 0), datetime.datetime(2017, 8, 24, 0, 0), datetime.datetime(2017, 8, 25, 0, 0), datetime.datetime(2017, 8, 28, 0, 0), datetime.datetime(2017, 8, 29, 0, 0), datetime.datetime(2017, 8, 30, 0, 0), datetime.datetime(2017, 8, 31, 0, 0), datetime.datetime(2017, 9, 1, 0, 0), datetime.datetime(2017, 9, 4, 0, 0), datetime.datetime(2017, 9, 5, 0, 0), datetime.datetime(2017, 9, 6, 0, 0), datetime.datetime(2017, 9, 7, 0, 0), datetime.datetime(2017, 9, 8, 0, 0), datetime.datetime(2017, 9, 11, 0, 0), datetime.datetime(2017, 9, 12, 0, 0), datetime.datetime(2017, 9, 13, 0, 0), datetime.datetime(2017, 9, 14, 0, 0), datetime.datetime(2017, 9, 15, 0, 0), datetime.datetime(2017, 9, 18, 0, 0), datetime.datetime(2017, 9, 19, 0, 0), datetime.datetime(2017, 9, 20, 0, 0), datetime.datetime(2017, 9, 21, 0, 0), datetime.datetime(2017, 9, 22, 0, 0), datetime.datetime(2017, 9, 25, 0, 0), datetime.datetime(2017, 9, 26, 0, 0), datetime.datetime(2017, 9, 27, 0, 0), datetime.datetime(2017, 9, 28, 0, 0), datetime.datetime(2017, 9, 29, 0, 0), datetime.datetime(2017, 10, 9, 0, 0), datetime.datetime(2017, 10, 10, 0, 0), datetime.datetime(2017, 10, 11, 0, 0), datetime.datetime(2017, 10, 12, 0, 0), datetime.datetime(2017, 10, 13, 0, 0), datetime.datetime(2017, 10, 16, 0, 0), datetime.datetime(2017, 10, 17, 0, 0), datetime.datetime(2017, 10, 18, 0, 0), datetime.datetime(2017, 10, 19, 0, 0), datetime.datetime(2017, 10, 20, 0, 0), datetime.datetime(2017, 10, 23, 0, 0), datetime.datetime(2017, 10, 24, 0, 0), datetime.datetime(2017, 10, 25, 0, 0), datetime.datetime(2017, 10, 26, 0, 0), datetime.datetime(2017, 10, 27, 0, 0), datetime.datetime(2017, 10, 30, 0, 0), datetime.datetime(2017, 10, 31, 0, 0), datetime.datetime(2017, 11, 1, 0, 0), datetime.datetime(2017, 11, 2, 0, 0), datetime.datetime(2017, 11, 3, 0, 0), datetime.datetime(2017, 11, 6, 0, 0), datetime.datetime(2017, 11, 7, 0, 0), datetime.datetime(2017, 11, 8, 0, 0), datetime.datetime(2017, 11, 9, 0, 0), datetime.datetime(2017, 11, 10, 0, 0), datetime.datetime(2017, 11, 13, 0, 0), datetime.datetime(2017, 11, 14, 0, 0), datetime.datetime(2017, 11, 15, 0, 0), datetime.datetime(2017, 11, 16, 0, 0), datetime.datetime(2017, 11, 17, 0, 0), datetime.datetime(2017, 11, 20, 0, 0), datetime.datetime(2017, 11, 21, 0, 0), datetime.datetime(2017, 11, 22, 0, 0), datetime.datetime(2017, 11, 23, 0, 0), datetime.datetime(2017, 11, 24, 0, 0), datetime.datetime(2017, 11, 27, 0, 0), datetime.datetime(2017, 11, 28, 0, 0), datetime.datetime(2017, 11, 29, 0, 0), datetime.datetime(2017, 11, 30, 0, 0), datetime.datetime(2017, 12, 1, 0, 0), datetime.datetime(2017, 12, 4, 0, 0), datetime.datetime(2017, 12, 5, 0, 0), datetime.datetime(2017, 12, 6, 0, 0), datetime.datetime(2017, 12, 7, 0, 0), datetime.datetime(2017, 12, 8, 0, 0), datetime.datetime(2017, 12, 11, 0, 0), datetime.datetime(2017, 12, 12, 0, 0), datetime.datetime(2017, 12, 13, 0, 0), datetime.datetime(2017, 12, 14, 0, 0), datetime.datetime(2017, 12, 15, 0, 0), datetime.datetime(2017, 12, 18, 0, 0), datetime.datetime(2017, 12, 19, 0, 0), datetime.datetime(2017, 12, 20, 0, 0), datetime.datetime(2017, 12, 21, 0, 0), datetime.datetime(2017, 12, 22, 0, 0), datetime.datetime(2017, 12, 25, 0, 0)]
星期一： 21.13%
星期二： 18.31%
星期三： 19.72%
星期四： 19.72%
星期五： 21.13%
星期六： 0.00%
星期日： 0.00%