In [84]:
import pandas as pd
import sklearn as sk
import seaborn as sb
import nltk
from nltk.corpus import gutenberg

In [168]:
ds = pd.read_csv("B:/Data_Mining_2/train_data.csv")
ds


Out[168]:
video_id uploader age category length views rate ratings comments 1 ... 11 12 13 14 15 16 17 18 19 20
0 LKh7zAJ4nwo TheReceptionist 653.0 Entertainment 424.0 13021.0 4.34 1305.0 744.0 NaN ... DjdA-5oKYFQ vmdPOOd6cxI uX81lMev6_o NaN W1Uo5DQTtzc heqocRij5P0 2ZgXx72XmoE 1tUDzOp10pk DH56yrIO5nI E-3zXq_r4w0
1 7D0Mf4Kn4Xk periurban 583.0 Music 201.0 6508.0 4.19 687.0 312.0 AvSK0qPw7EU ... e2k0h6tPvGc NaN VKFqqoeMdjw NSzrwv5MCwc y3IDp2n7B48 N1lkLaLJHlc NHB0a0xtLgU Eq4hGkIqBGw bdDskrr8jRY JngPWhfCb2M
2 n1cEq1C8oqQ Pipistrello 525.0 Comedy 125.0 1687.0 4.01 363.0 141.0 YfShxdbAJS8 ... eprHhmurMHg yLup8wjbSIo z5wDjq8o60c bWlPSLUT-6U bSPVayE0NhE F2_5KOnSsfI dsBTo5LExr0 gnpvEvuiFoQ 3nH5Tccz8EQ sEqCkwPmQ_w
3 OHkEzL4Unck ichannel 638.0 Comedy 299.0 8043.0 4.40 518.0 371.0 NaN ... eyUSTmEUQRg LQUV_XGzHmA nfBfC8bif1Y aiYwo5K0VWg N4EYgXReBzM FMuWYExDEJk Ml2NaXU6gms DVNwUKAuB3I mDjwzhc8dQ0 NyC_0Z6zoUk
4 #NAME? mrpitifulband 639.0 Music 287.0 7548.0 4.48 606.0 386.0 JsdCu9T47iY ... fmUwUURgsX0 eV2SdBITv8k sf-Ym_pFP6U 0vhVZQEzgcU lhWk9SXUjWI geiABCqmQ84 dDhCZVQf9po RZo5MisSTWo saBmFpuwmKA aVhSaa6aAOg
5 hFFH8DaOHQg istothehalfabee 592.0 Music 286.0 1759.0 4.45 539.0 244.0 GUgJKzEmsYI ... hFFH8DaOHQg 4qf7RSNCg40 oDIIOV4VKlA vEtM1q6gm9Q NaN NaN r89-fFx_tHU D6ABDEdhQLA 7TyH0ipgdtY jqNs_S0n7P8
6 LzHjIj3fpR8 Xelanderthomas 686.0 Comedy 168.0 4545.0 4.58 273.0 167.0 PlPynB10vP0 ... NaN Y_59kWK5W3s vta4RfQ2Z-I vWGA5iYgAOU 4WwVOWIqE80 kdZk1Wk7kSw NaN 4ALe2z---e0 NaN Qeeq5OoLGJ0
7 SDNkMu8ZT68 w00dy911 630.0 People & Blogs 186.0 10181.0 3.49 494.0 257.0 NaN ... rjnbgpPJUks NaN NaN NaN NaN NaN NaN NaN NaN NaN
8 PkGUU_ggO3k theresident 704.0 Entertainment 262.0 11235.0 3.85 247.0 280.0 K7unV366Qr4 ... PkGUU_ggO3k fbq2-jd5Dto p0lq5-8IDqY AfBxANiGnnU gVSzbvFnVRY NaN RyWz8hwGbY4 mh_MGyx9tgc JO1LTIFOkTw l9NJ04JiZj4
9 RX24KLBhwMI lemonette 697.0 People & Blogs 512.0 24149.0 4.22 315.0 474.0 NaN ... t60tW0WevkE dGM3k_4cNhE 2cKd9ERh5-8 EGWutOjVx4M sfG2rtAkAcg 9BBu5N0iFBg KVkseZR5coU 3uKOSjE79YA MwynZ8qTwXA j72VLPwzd_c
10 MEvoy_owET8 smpfilms 736.0 Travel & Places 921.0 109673.0 4.25 1181.0 774.0 Y5pMgbhyb18 ... YtX2nwowMtU W-s_e61hkys yxbLFd6Y38E UEXvMJo3ZAY srcg9xLjtuE NaN sCTbH-VP7mA JEiATJFBWO0 faDB-ToajhM 9aE4eMVeUEw
11 w6Pbyg_kcEk DrRek 388.0 Music 343.0 19096.0 3.78 376.0 290.0 loczWpMNAK4 ... 0I_5Kv7mZcE mI1N1YAFODk NaN irYf7cmmDok kKJl_FJviKI TS63Ae5p43E v_MCVw12iTQ xoTcHI2OcMg 4irUGiPe-EA rQdTSlLMKAA
12 Qehxjub5lyo jgrandin1 715.0 Entertainment 177.0 944984.0 4.50 3641.0 1943.0 wph_pmP9XC4 ... LzE2G4qaz9k EGHoPrUR3Xg DyccrmBwmXc L5vHXDXXHyU 62Opv2W8XJc EmioYVrE3y0 F-x_oA96NlI R_BwlXkWS_M pYCs_Rv2dfs G6f9zxSDgD8
13 sRnLdmvQBsI pdrop 735.0 Entertainment 188.0 67265.0 3.09 302.0 262.0 VNlmoPvcIcg ... NaN qRnY1myum-c THm4gpmcUgI rh2Gz5wiwvw fNaYQ4kM4FE -3bxyxMQg3Q yQQFZFl6fng uvl-j1hHdXs UMaHSkx_oUI SkfOvv6WR1E
14 xGn0q1zoibw FrontsidePromotions 706.0 Music 212.0 352964.0 4.29 1674.0 868.0 45EXfISJ7ME ... LJ48J8jUEvQ iFhg6APgibM NaN qatOFKmLoN4 V_eZzxlwOeA YknqUHvd8RI NaN ANze0PkfLPk kBUSfswmIEI NaN
15 zRVts7TFw-Y Philinthecircle 716.0 Film & Animation 309.0 607456.0 4.79 5711.0 3866.0 mxIRVfDMN30 ... s2UHrT1vWLs fMDqFn28jvc c83QQzQ7Flc NaN jH26XfCPg_Y sH3Gu8axlfY zOuOS0TemNM 2mOvPxUSPJQ U7iTWI4epMI qKKDzbi_84I
16 _5QUdvUhCZc lendavis 693.0 Travel & Places 105.0 819974.0 4.20 4007.0 2778.0 _hMfbzRsU0Q ... NaN 8fikJbiIldI UanGgNR1P3Q NaN NaN 7FRAOx129aY NaN l84oux3bnfg SmV9VeHCgNg 3oWFq1CeEa0
17 D6U6GxnvlE8 Karloze 678.0 Comedy 98.0 467849.0 3.65 2699.0 1431.0 m4QfvkszPUI ... 7kPyylUs7LM Ku4lo7Dwaqs NaN iVl76M0bzQc mwVLY6KRhXs gdhukVh2BpQ jePAAlNJvOI oxnI_liO4ew i1vk7kurbh0 Qi6ZYuVu3hc
18 OUi9-jqq_i0 ardentoctopus 650.0 Music 184.0 862371.0 4.58 5640.0 3132.0 yiKWisIhD6I ... 0gD_c8AHl4w KUBLQ7xeOD4 lacIXorMycY zGvuahsGhbo ZL4AFuCpuzE NaN amRtD-4aNyA jeUAmeXo8RE p7EG7i39F2w LzxAeqgd66Q
19 bqZauhidT1w bungloid 592.0 Film & Animation 28.0 374550.0 4.19 3588.0 1763.0 6T8DqGI__Xs ... NaN A61F0Z3td7w Cb1Pt7ygGFI YtlYriAEnXA HIsaAgmoAqE IQ5NSZ1SNLM kgnK9xU60GE NaN JI3Lh5wLeWk qEhZdQdJl_8
20 fQShwYqGqsw lonelygirl15 736.0 People & Blogs 133.0 151763.0 3.01 666.0 765.0 8YoxhsUMlgA ... fQShwYqGqsw X3ctuFCCF5k 113yn3sv0eo NaN vPUAf43vc-Q mUd0hcEnHiU aQWdqI1vd6o IAY5q60CmYY NaN ZllfQZCc2_M
21 mWzdp7Cg41w toshiaki1973 735.0 Entertainment 582.0 142699.0 3.45 148.0 146.0 KThI4-FdxXg ... hIbPgEyOGs4 Wp4ykVSBfts NaN yWGp-qmRtLU JOr19ALbF1I C0HLcNyRTZc AqKsXYnuN8s NaN XxCwlYQzDhw 1i3GX0Y9mf0
22 #NAME? KICKABOLAdotCOM 735.0 Sports 98.0 99425.0 4.36 101.0 168.0 N8jz_wEIo6o ... OPtkzcaPe1c cRbF2Aq87GE yvjfYe4PTEc YgWSRpcTZTY NaN 91_VUWLpx8Y sKnA7u-EatU Cl-pZtaY3Q0 NaN peqZZjKiuSo
23 H-ucblRMjuY nevrknw 735.0 People & Blogs 22.0 299935.0 2.67 640.0 213.0 J7d01xFC2GY ... 9jEZyxFs1C0 4psD5FlJCn8 8TNeXmXMWBY uAaQdthRYEw #NAME? 29taWVbP-uY NaN xajrbuaOp2E 2d5fjIDafH4 LOdSctUe6Gg
24 jDRENHHatx4 Silkworm17 736.0 Film & Animation 1444.0 105097.0 4.77 522.0 312.0 nWh_-SWqybE ... yWyq_40s5kQ ByeV9toSk4M _yEZIIqE_38 ySJw-tItRBM rqvVrCDmAL4 dFxo75eEZS4 HmWaXXLrRW0 NaN SySnQyTtBN0 fV1vhA8NI3M
25 ztIH6tc6Aa4 Pwned1555 735.0 Sports 421.0 61128.0 4.57 228.0 396.0 U-f-Nn5-uCc ... 6heT9_9Wm8M NaN G4gypcntwZo 4L8kjHfx6Zo F9qvkMq7EIE bIcgGP7WlFc D7C2YZOy9nA Zkx0USQXKrM yf-tLZCTKYw jLDOyxCDCtg
26 l69Vi5IDc0g Blendtec 736.0 Entertainment 88.0 86207.0 4.82 921.0 578.0 xx3ffJU2IS8 ... WJpprtKagLs H32iYY_E7BE jNPeSYmg4yg fNvseWjiWjs EfpS6-uIG84 rIvnGxYBlQw uJBX586Qp5s Cydp7a3WyKQ NW0fuDNOv4s NqYFdb6RDLQ
27 cn34xQU2M94 dellbhoy67 735.0 Sports 53.0 53556.0 4.45 112.0 155.0 WOh11lN5AW4 ... EPAqJ1aRjZ0 DpqQC8UKKgQ ONr4_md3x30 eu7S8xo9_pQ NaN B7U5_6sAdoI pCsvubJ-QXo NlDBgmCxadc bhSjOsqqsMY tLnCrjWACHU
28 aGEsRXlMT0s ysabellabrave 735.0 News & Politics 233.0 32406.0 4.42 674.0 579.0 NaN ... LSKgFqisXRo YUkIPRJBf5w twmaT9JWfmY OjnilzmjyG0 TCOKOvQiWFQ HgcEYqXO728 rRAVoyUCxz8 MPA-aySkb3M DC9uAN_2_tY eE7kjHlPm1Y
29 gZ_eaBgaDzY Kontrabandman 736.0 Comedy 54.0 76086.0 4.44 261.0 174.0 TjmOK4TTvwg ... x5THGTVjiQo uMoQO4_Faxw JMvMzQ4Vu-8 DxOjlATR-Lw GujYYYHYzHY ZedCXQhBF9I 2wQa6Qa-6Ew NaN 5C7v5vwSvIc CnXULd9fB3M
... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ...
26871 sLzH82ISe8c bigky226 621.0 Entertainment 9.0 2277.0 3.00 2.0 3.0 NaN ... hp8wcuhSJ0U liunMhW3EIw GuxDh7WtcBs P6f5EAvCS7Y GYGhmJD9LKc eciTd9okr00 K-x1NfQRSSA W7IxliAPjAk 4nRNYG_xM2U 8kOs3J0a2aI
26872 BrupD4_EFGA nippysblog 584.0 Pets & Animals 153.0 764.0 5.00 12.0 2.0 q8DZLE9Mxr0 ... DAxg95ISnoU KLzMKnrBVWE mLcT-7O6kdE dAFfu3dO8gk b55FELprjTo QCEnN8eNCeQ HKSr99d2IUA PlyH2ZrBWa0 WquwyOxHrlY K-x1NfQRSSA
26873 g76ql9lv3YI lonelygrl15 587.0 Comedy 44.0 3375.0 2.90 10.0 3.0 apLHtg9I1hM ... o4x-VW_rCSE Jx0_jk1isZw q8DZLE9Mxr0 HKSr99d2IUA K-x1NfQRSSA NaN J5MdBpaNbO0 4nRNYG_xM2U b55FELprjTo naRAHyZDdz4
26874 HadvMBpoVOY imerintoday 693.0 People & Blogs 62.0 171.0 0.00 0.0 0.0 p2314Qks2mw ... cQlvwTiJU0s 3fA6l0v1YO0 I2S03ZAfoGU NaN 8dLpFl_CI7c U_WwkH0qPHM 4GH9hP-AxCk oq0_V-zSx5s ROYdyKGfrZs HL-2p2mpFhY
26875 5pxIBqxlbKs Gerb1ll 584.0 Entertainment 136.0 2362.0 3.00 4.0 1.0 w4lMCVUbAyA ... NqfPKws2X50 lRyhh0PggoQ 7ump9ir4w-I NaN q8Clt5pv1TE ePM3mRAH1Oc i1Z6Eg3Suac p2314Qks2mw 8CnacaRbSho sd6VrXkx1Gs
26876 NqfPKws2X50 AlemaoGSI 730.0 Comedy 92.0 285.0 5.00 1.0 0.0 gPo1z6uCiBs ... 3nBPrMEzm5A dAFfu3dO8gk Jx0_jk1isZw ax58nnnNu2o K-x1NfQRSSA KLzMKnrBVWE J5MdBpaNbO0 5tog3M64glo WquwyOxHrlY b55FELprjTo
26877 gbUpMpJzT64 Eagle1118 585.0 Pets & Animals 29.0 9180.0 4.10 31.0 13.0 ftQL9RXLatI ... qnzA4daTAno xFCcN146WBw o-bficp7vDk Qp9GOXhuQQI NaN MK98G7yVcEg Oz2b_YQtVws K-x1NfQRSSA w4lMCVUbAyA NaN
26878 1eU8fpDEvxs funnyman222 616.0 Comedy 34.0 6327.0 4.75 20.0 14.0 DKJjMEmtDOc ... y1M7xgQZD2Q SJOEt0A2dLk gbUpMpJzT64 k0Dys4XOnvs kcCdp6iMMYA L5vXlcOSBrk gf0VbW1dWRk o4x-VW_rCSE tHJ2OGyGMgY gPo1z6uCiBs
26879 bfUvtLonV6k sandsquid74 654.0 Entertainment 112.0 428.0 3.00 2.0 0.0 EbpO-8xv7sU ... 0TQEGbFGU5s R7BgOguk8rY Oz2b_YQtVws NaN 2r0ZEJulQu4 w4lMCVUbAyA EvYkYGeXcMw Ui5v3APjmp8 NaN hp8wcuhSJ0U
26880 gNJf_bvAVIY david12588 677.0 Comedy 60.0 154.0 3.00 2.0 2.0 b55FELprjTo ... o4x-VW_rCSE xLaHpl7CmRU 80TKvIsuf9A eciTd9okr00 8kOs3J0a2aI M4RErUV9Jv4 P6f5EAvCS7Y K-x1NfQRSSA GYGhmJD9LKc W7IxliAPjAk
26881 Oe93ZMnzuYI brienicole 676.0 People & Blogs 80.0 425.0 5.00 1.0 2.0 HadvMBpoVOY ... g-wMxnuioMQ 9PRWMOBEt88 D0V-4zYz0Kg WvrtIXdbivA 6HCgOcdmC9A ca3_xCg6twk lC3cLI_wT7k sd_FXdAQKac 2fN4I26yA-k AZjKpqJrMSQ
26882 dgz2xKOwoMY lctrom2003 682.0 Comedy 35.0 108.0 0.00 0.0 0.0 NaN ... u1heFIhM1P4 52VQjyI4nH0 2sPmC3x_SbM NaN uRx5Inn4PGo eN0D0magxGk xOf1u3W1irY El0VTvExuqQ j2MeNvPTpB4 QCEnN8eNCeQ
26883 2sIkVJIdiFw bkendig 672.0 Comedy 20.0 1926.0 4.86 7.0 2.0 NaN ... cGn9C3EhP2Q tfQXYWOo-hM NaN G20l4TPKkaI UHjYQc1651A fCoFeJQq7IU R2ymZR9L_Pg RbxZXJT0QNQ NaN q0a9qPkYHBg
26884 nZsMLewmjv8 Seaside7 581.0 Comedy 38.0 4946.0 5.00 6.0 3.0 joTAAizhecg ... 8dLpFl_CI7c s0XY3c2nRvk t8WXgFWHOx4 G-en6ourxhc z3cBzn6NCcQ NaN foG-8iS7zx4 sLrOcGuOI5o ouVuTEATV2I 3MM2bYgSqVo
26885 wI0nB0eR8yA speedyjoe3 585.0 Pets & Animals 30.0 47520.0 2.96 51.0 19.0 dAFfu3dO8gk ... JYRKSydlnqc NaN Bl3T-Nsb2AQ cNt29huGNoc NaN b55FELprjTo cQsr2ddw2mc GrdglA2t1-c gDzJjznJciE w4lMCVUbAyA
26886 cQsr2ddw2mc serglinkk 583.0 Howto & DIY 118.0 18111.0 4.43 30.0 19.0 EbpO-8xv7sU ... o4x-VW_rCSE Oz2b_YQtVws QDz3DjM3MUA gwTLFJFRW0Q Ui5v3APjmp8 Qp9GOXhuQQI RuLidm9UKd8 8kOs3J0a2aI VkIHdw8Jrnk 7ump9ir4w-I
26887 gwTLFJFRW0Q maikikin 609.0 Entertainment 85.0 4315.0 5.00 4.0 3.0 74MKKhrGoM0 ... o4x-VW_rCSE Oz2b_YQtVws Bl3T-Nsb2AQ cNt29huGNoc Ui5v3APjmp8 Qp9GOXhuQQI NaN NaN VkIHdw8Jrnk 7ump9ir4w-I
26888 RuLidm9UKd8 violakevo 583.0 Pets & Animals 94.0 7342.0 3.82 44.0 11.0 HKSr99d2IUA ... o4x-VW_rCSE b55FELprjTo Jx0_jk1isZw VkIHdw8Jrnk cNt29huGNoc naRAHyZDdz4 igecQ61MPP4 dAFfu3dO8gk 7ump9ir4w-I cQsr2ddw2mc
26889 QDz3DjM3MUA RobotsnStuff 602.0 Pets & Animals 246.0 2380.0 3.29 7.0 1.0 CznjykhygFA ... 87HVROW7nrU NaN Y2YcaADdAb0 MpHAGodq7wU cA59Sa-MCJo 4_QTKIE-wKg NaN PBal0f30I6M 44U0OAEd4N4 HanVUshC4gU
26890 Bl3T-Nsb2AQ karunaunddaniel 632.0 Entertainment 134.0 8791.0 4.75 16.0 13.0 NaN ... FtraOormiBw NaN SJOEt0A2dLk VkIHdw8Jrnk lo_8F9-i1qw Qp9GOXhuQQI NaN xFCcN146WBw HQ91yvnbtZI o4x-VW_rCSE
26891 gcgV8nybXnw delic8genius 612.0 Entertainment 133.0 7800.0 4.75 4.0 6.0 egrqWzX9mWA ... o4x-VW_rCSE 7_vYEJ2-YL8 CPw1veBbACA gOSHeZ0wtpA 5ZYMAI5WpMQ 8HXi2szrE_I PlyH2ZrBWa0 5bKhj6-rtdI kYhsD6um15Q I1q7cKVtUe4
26892 572_H5wRROI miamidesigner 581.0 Gadgets & Games 112.0 3922.0 3.67 6.0 2.0 QDz3DjM3MUA ... yY_Hwih0Ctk NaN naRAHyZDdz4 7ump9ir4w-I o4x-VW_rCSE GrdglA2t1-c cNt29huGNoc igecQ61MPP4 hp8wcuhSJ0U w4lMCVUbAyA
26893 0TQEGbFGU5s ScottPJones 606.0 Comedy 53.0 89630.0 4.44 201.0 110.0 UA-Ixo5DlfY ... PLy48NlrWFw dAFfu3dO8gk gPo1z6uCiBs QCEnN8eNCeQ WquwyOxHrlY KLzMKnrBVWE b55FELprjTo Jx0_jk1isZw o4x-VW_rCSE K-x1NfQRSSA
26894 cZ5r4-TdHrQ ScottPJones 606.0 Comedy 58.0 27205.0 4.31 61.0 55.0 UA-Ixo5DlfY ... 0TQEGbFGU5s dAFfu3dO8gk gPo1z6uCiBs QCEnN8eNCeQ WquwyOxHrlY KLzMKnrBVWE b55FELprjTo Jx0_jk1isZw o4x-VW_rCSE K-x1NfQRSSA
26895 MK98G7yVcEg JangLeeYen 582.0 Pets & Animals 82.0 7486.0 4.56 18.0 12.0 74MKKhrGoM0 ... e2MoIohV66Q wI0nB0eR8yA K-x1NfQRSSA o4x-VW_rCSE eJ-kOEX0rrA Oz2b_YQtVws NaN Qp9GOXhuQQI gOSHeZ0wtpA wNMT_B6j3jw
26896 XgM54XCnM6E obuengineer 649.0 Comedy 14.0 469.0 3.00 1.0 0.0 8kOs3J0a2aI ... 9PRWMOBEt88 cNt29huGNoc NaN Ui5v3APjmp8 w4lMCVUbAyA Oz2b_YQtVws 7ump9ir4w-I DKJjMEmtDOc o4x-VW_rCSE NaN
26897 Mh8bS6EdyrM srleung88 653.0 Entertainment 67.0 273.0 0.00 0.0 0.0 BKKSvfc5qe0 ... wybJtnXIOJg pNlKmYvcs8g aGGp65WW9FI 2BiU9AHH9Wc Exf2sqOoRwg t2o5loLMzMQ 0Da0tYKdJaw -dkvR6n-3pQ oPsCG_o8b_k JfLRZwepszQ
26898 lsbcVYxVuWI audacityx 663.0 Comedy 61.0 5805.0 4.38 13.0 4.0 dAFfu3dO8gk ... o4x-VW_rCSE Axle4LsOtIw NaN DkyaWgCm6xQ naRAHyZDdz4 K-x1NfQRSSA NaN NaN g76ql9lv3YI Xl07P8ML1kw
26899 8c1mKLbVtFo crossover37 584.0 Comedy 13.0 7339.0 3.00 4.0 6.0 CPw1veBbACA ... a_RKaDGNQ98 Oz2b_YQtVws 5F6vRPa3x38 nZsMLewmjv8 xr_SXGt4bko pSDaMhOHDkc liunMhW3EIw XgM54XCnM6E QCEnN8eNCeQ PlyH2ZrBWa0
26900 VBGue5j_sUo dchulito 682.0 Entertainment 42.0 248.0 0.00 0.0 0.0 KQjtol9XEnM ... Ct4TWq5fd-E 3ABX2-4RTrY tUS2f3Y9Y-U Hf5CmNhwpSU ObmmQGv_Up8 g9OqJDsZxEo NaN YJnZ0_nghk8 vDhMiJzkKfk N_z5N4skaIA

26901 rows × 29 columns


In [86]:
# ds = ds.drop(ds.columns[[1,2,4,5,6,7,8]], axis=1)

In [169]:
df = pd.DataFrame(columns=['video_id','Entertainment','Music', 'Comedy', 'People & Blogs',
       'Travel & Places', 'Film & Animation', 'Sports', 'News & Politics',
       'Gadgets & Games', 'Howto & DIY', 'Pets & Animals',
       'Autos & Vehicles', ' UNA ','nan'])
i=0
for index,rows in ds.iterrows():
    for label in range(0,20):
        if(label == 0) : 
            toquery = rows.video_id
        else :
            toquery = rows[str(label)]
        if(str(toquery)!='nan' and str(toquery)!='#NAME?'):
            repeat_row = df[df.video_id==toquery]
            if(repeat_row.size == 0):
                new_row = {'video_id' : str(toquery),'Entertainment': 0,'Music': 0, 'Comedy': 0, 'People & Blogs': 0,
                'Travel & Places': 0, 'Film & Animation': 0, 'Sports': 0, 'News & Politics': 0,
                'Gadgets & Games': 0, 'Howto & DIY': 0, 'Pets & Animals': 0,
                'Autos & Vehicles': 0, ' UNA ': 0,'nan' : 0}
                s = pd.Series(new_row)
                s[str(rows.category)] += 1
                df.loc[i] = s
                i+=1
            else :
                repeat_row[str(rows.category)] += 1
                df.update(repeat_row)


C:\Program Files\Anaconda2\lib\site-packages\ipykernel\__main__.py:24: SettingWithCopyWarning: 
A value is trying to be set on a copy of a slice from a DataFrame.
Try using .loc[row_indexer,col_indexer] = value instead

See the caveats in the documentation: http://pandas.pydata.org/pandas-docs/stable/indexing.html#indexing-view-versus-copy

In [170]:
copied = df.copy()
copied


Out[170]:
video_id Entertainment Music Comedy People & Blogs Travel & Places Film & Animation Sports News & Politics Gadgets & Games Howto & DIY Pets & Animals Autos & Vehicles UNA nan
0 LKh7zAJ4nwo 2.0 0.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
1 xiDqywcDQRM 19.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
2 LGVU5DsezE0 31.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
3 NxTDlnOuybo 1.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
4 KRHfMQqSHpk 11.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
5 1TCeoRPg5dE 2.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
6 _XIuvoH6rUg 22.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
7 c-8VuICzXtU 2.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
8 -7ClGo-YgZ0 24.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
9 DjdA-5oKYFQ 2.0 0.0 1.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
10 vmdPOOd6cxI 5.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
11 uX81lMev6_o 21.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
12 W1Uo5DQTtzc 5.0 1.0 2.0 1.0 0.0 1.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0
13 heqocRij5P0 14.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
14 2ZgXx72XmoE 7.0 0.0 1.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
15 1tUDzOp10pk 18.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
16 DH56yrIO5nI 2.0 0.0 0.0 1.0 0.0 0.0 6.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
17 7D0Mf4Kn4Xk 1.0 11.0 1.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
18 AvSK0qPw7EU 0.0 5.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
19 WX5KLMqY4bM 0.0 4.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
20 zpTorUhCd8Y 0.0 7.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
21 AK8Wtfwe-1k 0.0 9.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
22 eUIfRyrqwp8 0.0 7.0 1.0 4.0 0.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0
23 KQaUvH5oiO4 0.0 7.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 1.0 0.0 0.0
24 VqpnWBo-R4E 0.0 8.0 1.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
25 DlRodd4s86s 0.0 9.0 1.0 7.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
26 e2k0h6tPvGc 0.0 12.0 1.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
27 VKFqqoeMdjw 0.0 7.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
28 NSzrwv5MCwc 0.0 7.0 0.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
29 y3IDp2n7B48 1.0 8.0 0.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
... ... ... ... ... ... ... ... ... ... ... ... ... ... ... ...
207595 agUG9Ha19uI 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207596 JPKb2SrbZfE 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207597 ARdoZh16uzM 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207598 wybJtnXIOJg 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207599 pNlKmYvcs8g 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207600 aGGp65WW9FI 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207601 2BiU9AHH9Wc 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207602 Exf2sqOoRwg 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207603 t2o5loLMzMQ 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207604 0Da0tYKdJaw 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207605 -dkvR6n-3pQ 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207606 oPsCG_o8b_k 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207607 xkGlEsI_FdM 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207608 KQjtol9XEnM 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207609 f5ghRvEjgWU 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207610 iYVn2KU5b1E 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207611 PEDDUvYfvwg 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207612 tzwdwOU71mE 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207613 brZq-Zep4h8 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207614 2zcfvF52qbU 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207615 3-Zgjh4gauA 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207616 mGf5OqUX-GU 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207617 Ct4TWq5fd-E 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207618 3ABX2-4RTrY 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207619 tUS2f3Y9Y-U 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207620 Hf5CmNhwpSU 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207621 ObmmQGv_Up8 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207622 g9OqJDsZxEo 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207623 YJnZ0_nghk8 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207624 vDhMiJzkKfk 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0

207625 rows × 15 columns


In [171]:
del df['video_id']
df


Out[171]:
Entertainment Music Comedy People & Blogs Travel & Places Film & Animation Sports News & Politics Gadgets & Games Howto & DIY Pets & Animals Autos & Vehicles UNA nan
0 2.0 0.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
1 19.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
2 31.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
3 1.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
4 11.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
5 2.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
6 22.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
7 2.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
8 24.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
9 2.0 0.0 1.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
10 5.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
11 21.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
12 5.0 1.0 2.0 1.0 0.0 1.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0
13 14.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
14 7.0 0.0 1.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
15 18.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
16 2.0 0.0 0.0 1.0 0.0 0.0 6.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
17 1.0 11.0 1.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
18 0.0 5.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
19 0.0 4.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
20 0.0 7.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
21 0.0 9.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
22 0.0 7.0 1.0 4.0 0.0 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0
23 0.0 7.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 1.0 0.0 0.0
24 0.0 8.0 1.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
25 0.0 9.0 1.0 7.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
26 0.0 12.0 1.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
27 0.0 7.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
28 0.0 7.0 0.0 2.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
29 1.0 8.0 0.0 3.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
... ... ... ... ... ... ... ... ... ... ... ... ... ... ...
207595 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207596 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207597 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207598 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207599 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207600 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207601 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207602 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207603 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207604 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207605 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207606 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207607 0.0 0.0 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207608 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207609 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207610 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207611 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207612 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207613 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207614 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207615 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207616 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207617 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207618 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207619 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207620 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207621 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207622 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207623 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0
207624 1.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0 0.0

207625 rows × 14 columns


In [172]:
answer = pd.DataFrame(columns=['video_id','category'])
i=0
for index,rows in df.iterrows():
    new_row = {'category' : rows.idxmax()}
    s = pd.Series(new_row)
    answer.loc[i] = s
    i+=1
answer['video_id'] = copied.video_id
answer


Out[172]:
video_id category
0 LKh7zAJ4nwo Comedy
1 xiDqywcDQRM Entertainment
2 LGVU5DsezE0 Entertainment
3 NxTDlnOuybo Entertainment
4 KRHfMQqSHpk Entertainment
5 1TCeoRPg5dE Entertainment
6 _XIuvoH6rUg Entertainment
7 c-8VuICzXtU Entertainment
8 -7ClGo-YgZ0 Entertainment
9 DjdA-5oKYFQ Entertainment
10 vmdPOOd6cxI Entertainment
11 uX81lMev6_o Entertainment
12 W1Uo5DQTtzc Entertainment
13 heqocRij5P0 Entertainment
14 2ZgXx72XmoE Entertainment
15 1tUDzOp10pk Entertainment
16 DH56yrIO5nI Sports
17 7D0Mf4Kn4Xk Music
18 AvSK0qPw7EU Music
19 WX5KLMqY4bM Music
20 zpTorUhCd8Y Music
21 AK8Wtfwe-1k Music
22 eUIfRyrqwp8 Music
23 KQaUvH5oiO4 Music
24 VqpnWBo-R4E Music
25 DlRodd4s86s Music
26 e2k0h6tPvGc Music
27 VKFqqoeMdjw Music
28 NSzrwv5MCwc Music
29 y3IDp2n7B48 Music
... ... ...
207595 agUG9Ha19uI Entertainment
207596 JPKb2SrbZfE Entertainment
207597 ARdoZh16uzM Entertainment
207598 wybJtnXIOJg Entertainment
207599 pNlKmYvcs8g Entertainment
207600 aGGp65WW9FI Entertainment
207601 2BiU9AHH9Wc Entertainment
207602 Exf2sqOoRwg Entertainment
207603 t2o5loLMzMQ Entertainment
207604 0Da0tYKdJaw Entertainment
207605 -dkvR6n-3pQ Entertainment
207606 oPsCG_o8b_k Entertainment
207607 xkGlEsI_FdM Comedy
207608 KQjtol9XEnM Entertainment
207609 f5ghRvEjgWU Entertainment
207610 iYVn2KU5b1E Entertainment
207611 PEDDUvYfvwg Entertainment
207612 tzwdwOU71mE Entertainment
207613 brZq-Zep4h8 Entertainment
207614 2zcfvF52qbU Entertainment
207615 3-Zgjh4gauA Entertainment
207616 mGf5OqUX-GU Entertainment
207617 Ct4TWq5fd-E Entertainment
207618 3ABX2-4RTrY Entertainment
207619 tUS2f3Y9Y-U Entertainment
207620 Hf5CmNhwpSU Entertainment
207621 ObmmQGv_Up8 Entertainment
207622 g9OqJDsZxEo Entertainment
207623 YJnZ0_nghk8 Entertainment
207624 vDhMiJzkKfk Entertainment

207625 rows × 2 columns


In [236]:
answer.to_csv('answer.csv',index=False)

In [282]:
test_data = pd.read_csv("B:/Data_Mining_2/test_data.csv")
test_data.insert(3,'edge_present',0)

In [283]:
test_data


Out[283]:
source target edge_id edge_present
0 lUjVSKfdCWU yL2faZW8Htc 0 0
1 mlXkoUV3zqQ d0X6H_31tgc 1 0
2 OpIyo7esbtQ SJ1n5ecSylU 2 0
3 x4T24P0d-jE YRK9MD11_q8 3 0
4 AI62Y4ijmgg G_XLwK3ItuY 4 0
5 QgJ4onNY_8I vK1TMw4rQ24 5 0
6 tx-RpOlM7H4 nJi56F1_yE8 6 0
7 lCqUtK0WgBM wElZmFQjKsk 7 0
8 ckLQF7QFmWc BDzzf6PBr80 8 0
9 RPWi414S69M mnkRqUN_c7s 9 0
10 -1hkThIjUvE CZo89Xz6OA0 10 0
11 Agm2-OqTol4 pk6g1TVysro 11 0
12 bQEYQbza2G0 OWuULYNE_1Q 12 0
13 6jOm4Q8_6nk Zal_U_RXLa8 13 0
14 8IpLlBhhVJs BmJCB3yuKDk 14 0
15 fw_ebDka2IE J5YpBdZC69w 15 0
16 F8Mw_zVuk2I jEFxXdR8YV0 16 0
17 nOMDGx2xYw0 D_NQ5jQBtkE 17 0
18 kp6Eh4XGlWA QyTKCbmiBqg 18 0
19 AagSqXud2Hw HRpNXTx__F4 19 0
20 xKzxCgAJrxw VgNUtpeen40 20 0
21 YKZFxW4pvIo gdMtLS_mV38 21 0
22 q3G9fDO5PYE WeamNPxW6iA 22 0
23 KO9mMS_VJNU JYkxcNy5I6o 23 0
24 gz19ZP9roRQ tgjhjtNrwt0 24 0
25 c2-abqUxNLU PlQ_H8X9tvw 25 0
26 CkOiCPswOEQ A_cqtFrCs3U 26 0
27 4vuQeCy3wEE Klw52tQNuP0 27 0
28 HTiQhrdmgW4 tXtacme8jFk 28 0
29 DVzY1EtzRJ8 44Wo7oCk6nM 29 0
... ... ... ... ...
99970 8qVay9cj7Ps J84kpsPEB94 99970 0
99971 FUB_XU_1P8w FSKlzaPWJbI 99971 0
99972 N5m775HUmSg a7otD_t0UmU 99972 0
99973 Ykx7czlmUXU evuk2-zAZmQ 99973 0
99974 9dBl558AaiM xXrnKGSShMk 99974 0
99975 4i92FplrHUI 3YGPll7s3vg 99975 0
99976 vygEVxYBXDU o9WbVQ9jJS4 99976 0
99977 _0Wo6KVicic AkYi6413QFI 99977 0
99978 FPK6aHd-rE4 J0QsxjswWnc 99978 0
99979 FrkK3kbY4yA 7t6QmmF7okY 99979 0
99980 rqmIyBtHOzM oT0Zea3pkLE 99980 0
99981 jUNTZyTFF1w OuCq2mwGUpY 99981 0
99982 ac_qWfAZKJY Jvubu0xTSnM 99982 0
99983 SWyWKNW75Ss XlJ_g10u4G8 99983 0
99984 yJUeOqL9nKY pH7X2NDsZLk 99984 0
99985 wnQ96nuNsTk 96Rr8SQ2AR8 99985 0
99986 Qg4ePPgLPLU x9g2emM7lDU 99986 0
99987 iU7HE6MkMqY qyzqkF3npZA 99987 0
99988 Qb8tofqG-XY FZTKDFFQgA0 99988 0
99989 MYTJooXcG6g wS5gh0NXuaY 99989 0
99990 z3lXMQQtt8w FwKQi7HB_Gw 99990 0
99991 KRUV7iWfXgk GSUBk6VcGKw 99991 0
99992 UGYCY48Tugw ZtwXwqlIIsw 99992 0
99993 4w2r4tdq848 OLcIrdxh31k 99993 0
99994 Sq8UaSDrTUQ fqBT8XISWbw 99994 0
99995 hNkEIqFxyrU FSpwiGFyvo8 99995 0
99996 fFBC5aa1wFw _uzuGuO0WnI 99996 0
99997 QZudcoWa-6k P9B3qr6uzHA 99997 0
99998 r__Sj6fNaRc KL9hx2MFZAY 99998 0
99999 KdCe2Zrr2sQ AOI5y_0JXVc 99999 0

100000 rows × 4 columns


In [284]:
for index,rows in test_data.iterrows():
    toquery1 = rows.source
    repeat_row1 = answer[answer.video_id==toquery1]
    toquery2 = rows.target
    repeat_row2 = answer[answer.video_id==toquery2]
    if(repeat_row1.size != 0 and repeat_row2.size != 0) :
        if(str(repeat_row1.category.values[0]) == str(repeat_row2.category.values[0])):
            rows.edge_present = 1
            test_data.loc[index] = rows

In [285]:
test_data


Out[285]:
source target edge_id edge_present
0 lUjVSKfdCWU yL2faZW8Htc 0 0
1 mlXkoUV3zqQ d0X6H_31tgc 1 0
2 OpIyo7esbtQ SJ1n5ecSylU 2 0
3 x4T24P0d-jE YRK9MD11_q8 3 0
4 AI62Y4ijmgg G_XLwK3ItuY 4 1
5 QgJ4onNY_8I vK1TMw4rQ24 5 0
6 tx-RpOlM7H4 nJi56F1_yE8 6 0
7 lCqUtK0WgBM wElZmFQjKsk 7 0
8 ckLQF7QFmWc BDzzf6PBr80 8 1
9 RPWi414S69M mnkRqUN_c7s 9 0
10 -1hkThIjUvE CZo89Xz6OA0 10 0
11 Agm2-OqTol4 pk6g1TVysro 11 0
12 bQEYQbza2G0 OWuULYNE_1Q 12 0
13 6jOm4Q8_6nk Zal_U_RXLa8 13 0
14 8IpLlBhhVJs BmJCB3yuKDk 14 1
15 fw_ebDka2IE J5YpBdZC69w 15 0
16 F8Mw_zVuk2I jEFxXdR8YV0 16 0
17 nOMDGx2xYw0 D_NQ5jQBtkE 17 0
18 kp6Eh4XGlWA QyTKCbmiBqg 18 1
19 AagSqXud2Hw HRpNXTx__F4 19 1
20 xKzxCgAJrxw VgNUtpeen40 20 0
21 YKZFxW4pvIo gdMtLS_mV38 21 1
22 q3G9fDO5PYE WeamNPxW6iA 22 0
23 KO9mMS_VJNU JYkxcNy5I6o 23 0
24 gz19ZP9roRQ tgjhjtNrwt0 24 0
25 c2-abqUxNLU PlQ_H8X9tvw 25 0
26 CkOiCPswOEQ A_cqtFrCs3U 26 0
27 4vuQeCy3wEE Klw52tQNuP0 27 1
28 HTiQhrdmgW4 tXtacme8jFk 28 0
29 DVzY1EtzRJ8 44Wo7oCk6nM 29 0
... ... ... ... ...
99970 8qVay9cj7Ps J84kpsPEB94 99970 0
99971 FUB_XU_1P8w FSKlzaPWJbI 99971 0
99972 N5m775HUmSg a7otD_t0UmU 99972 0
99973 Ykx7czlmUXU evuk2-zAZmQ 99973 0
99974 9dBl558AaiM xXrnKGSShMk 99974 0
99975 4i92FplrHUI 3YGPll7s3vg 99975 0
99976 vygEVxYBXDU o9WbVQ9jJS4 99976 1
99977 _0Wo6KVicic AkYi6413QFI 99977 0
99978 FPK6aHd-rE4 J0QsxjswWnc 99978 1
99979 FrkK3kbY4yA 7t6QmmF7okY 99979 1
99980 rqmIyBtHOzM oT0Zea3pkLE 99980 0
99981 jUNTZyTFF1w OuCq2mwGUpY 99981 0
99982 ac_qWfAZKJY Jvubu0xTSnM 99982 0
99983 SWyWKNW75Ss XlJ_g10u4G8 99983 0
99984 yJUeOqL9nKY pH7X2NDsZLk 99984 0
99985 wnQ96nuNsTk 96Rr8SQ2AR8 99985 0
99986 Qg4ePPgLPLU x9g2emM7lDU 99986 0
99987 iU7HE6MkMqY qyzqkF3npZA 99987 1
99988 Qb8tofqG-XY FZTKDFFQgA0 99988 0
99989 MYTJooXcG6g wS5gh0NXuaY 99989 0
99990 z3lXMQQtt8w FwKQi7HB_Gw 99990 0
99991 KRUV7iWfXgk GSUBk6VcGKw 99991 0
99992 UGYCY48Tugw ZtwXwqlIIsw 99992 0
99993 4w2r4tdq848 OLcIrdxh31k 99993 0
99994 Sq8UaSDrTUQ fqBT8XISWbw 99994 0
99995 hNkEIqFxyrU FSpwiGFyvo8 99995 1
99996 fFBC5aa1wFw _uzuGuO0WnI 99996 0
99997 QZudcoWa-6k P9B3qr6uzHA 99997 0
99998 r__Sj6fNaRc KL9hx2MFZAY 99998 0
99999 KdCe2Zrr2sQ AOI5y_0JXVc 99999 0

100000 rows × 4 columns


In [286]:
del test_data['source']
del test_data['target']
test_data


Out[286]:
edge_id edge_present
0 0 0
1 1 0
2 2 0
3 3 0
4 4 1
5 5 0
6 6 0
7 7 0
8 8 1
9 9 0
10 10 0
11 11 0
12 12 0
13 13 0
14 14 1
15 15 0
16 16 0
17 17 0
18 18 1
19 19 1
20 20 0
21 21 1
22 22 0
23 23 0
24 24 0
25 25 0
26 26 0
27 27 1
28 28 0
29 29 0
... ... ...
99970 99970 0
99971 99971 0
99972 99972 0
99973 99973 0
99974 99974 0
99975 99975 0
99976 99976 1
99977 99977 0
99978 99978 1
99979 99979 1
99980 99980 0
99981 99981 0
99982 99982 0
99983 99983 0
99984 99984 0
99985 99985 0
99986 99986 0
99987 99987 1
99988 99988 0
99989 99989 0
99990 99990 0
99991 99991 0
99992 99992 0
99993 99993 0
99994 99994 0
99995 99995 1
99996 99996 0
99997 99997 0
99998 99998 0
99999 99999 0

100000 rows × 2 columns


In [287]:
test_data.to_csv('submission2.csv',index=False)