Advertisement
Guest User

Untitled

a guest
Sep 19th, 2017
55
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.87 KB | None | 0 0
  1. import numpy as np
  2. import pandas as pd
  3. import matplotlib.pyplot as plt
  4.  
  5. x = np.arange(10)
  6. x = np.concatenate((x,x))
  7. y = []
  8. for i in range(2):
  9. y.append(np.random.random_integers(0,10,20))
  10.  
  11. d = {'A': [(x[i], y[0][i]) for i in range(20)],
  12. 'B': [(x[i], y[1][i]) for i in range(20)]}
  13. df = pd.DataFrame(d, index = list('aaaaaaaaaabbbbbbbbbb'))
  14.  
  15. A B
  16. a (0, 2) (0, 10)
  17. a (1, 0) (1, 8)
  18. a (2, 3) (2, 8)
  19. a (3, 7) (3, 8)
  20. a (4, 8) (4, 10)
  21. a (5, 2) (5, 0)
  22. a (6, 1) (6, 4)
  23. a (7, 3) (7, 9)
  24. a (8, 4) (8, 4)
  25. a (9, 4) (9, 10)
  26. b (0, 0) (0, 3)
  27. b (1, 2) (1, 10)
  28. b (2, 8) (2, 3)
  29. b (3, 1) (3, 7)
  30. b (4, 6) (4, 1)
  31. b (5, 8) (5, 3)
  32. b (6, 1) (6, 4)
  33. b (7, 1) (7, 1)
  34. b (8, 2) (8, 7)
  35. b (9, 9) (9, 3)
  36.  
  37. df['A.x'], df['A.y'] = zip(*df.A)
  38. df['B.x'], df['B.y'] = zip(*df.B)
  39.  
  40. >>> df.head()
  41. A B A.x A.y B.x B.y
  42. a (0, 6) (0, 0) 0 6 0 0
  43. a (1, 8) (1, 4) 1 8 1 4
  44. a (2, 8) (2, 5) 2 8 2 5
  45. a (3, 5) (3, 2) 3 5 3 2
  46. a (4, 2) (4, 4) 4 2 4 4
  47.  
  48. df['a1'], df['a2'] = df['A'].str[0], df['A'].str[1]
  49. df['b1'], df['b2'] = df['B'].str[0], df['B'].str[1]
  50.  
  51. print (df)
  52. A B a1 a2 b1 b2
  53. a (0, 5) (0, 1) 0 5 0 1
  54. a (1, 0) (1, 5) 1 0 1 5
  55. a (2, 3) (2, 9) 2 3 2 9
  56. a (3, 3) (3, 8) 3 3 3 8
  57. a (4, 7) (4, 9) 4 7 4 9
  58. a (5, 9) (5, 4) 5 9 5 4
  59. a (6, 3) (6, 3) 6 3 6 3
  60. a (7, 5) (7, 0) 7 5 7 0
  61. a (8, 2) (8, 3) 8 2 8 3
  62. a (9, 4) (9, 5) 9 4 9 5
  63. b (0, 7) (0, 0) 0 7 0 0
  64. b (1, 6) (1, 2) 1 6 1 2
  65. b (2, 8) (2, 3) 2 8 2 3
  66. b (3, 8) (3, 8) 3 8 3 8
  67. b (4, 10) (4, 1) 4 10 4 1
  68. b (5, 1) (5, 3) 5 1 5 3
  69. b (6, 6) (6, 3) 6 6 6 3
  70. b (7, 7) (7, 3) 7 7 7 3
  71. b (8, 7) (8, 7) 8 7 8 7
  72. b (9, 8) (9, 0) 9 8 9 0
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement