可视化向量运算with numpy & matplotlib

VIRUS_2091

import numpy as n
import matplotlib.pyplot as m

vec_a = n.array([1,1])
vec_b = n.array([-1,1])

vec_sum = vec_a + vec_b
vec_dif = vec_a - vec_b
vec_tim = vec_a * 0.5

print(vec_a)
print(vec_b)
print(vec_sum)
print(vec_dif)

m.quiver(0,0,vec_a[0],vec_a[1],
        angles = 'xy',scale_units = 'xy',
        scale = 1,color = 'red',label = 'a')

m.quiver(0,0,vec_b[0],vec_b[1],
        angles = 'xy',scale_units = 'xy',
        scale = 1,color = 'blue',label = 'b')

m.quiver(0,0,vec_sum[0],vec_sum[1],
        angles = 'xy',scale_units = 'xy',
        scale = 1,color = 'lime',label = 'a+b')

m.quiver(0,0,vec_dif[0],vec_dif[1],
        angles = 'xy',scale_units = 'xy',
        scale = 1,color = 'violet',label = 'a-b')

m.quiver(0,0,vec_tim[0],vec_tim[1],
        angles = 'xy',scale_units = 'xy',
        scale = 1,color = 'yellow',label = '0.5a')

m.title('Vector Samples')

m.xticks(n.arange(-2, 3, 1))  

m.yticks(n.arange(0, 4, 1))

m.figure(figsize = (5,5))

m.show()

FishC

问题主要出在vec_a和vec_b没有给具体数值，n.array()会报错；另外quiver的用法不太对，前两个参数是起点坐标，后两个参数是向量的U、V分量，不能直接把整个向量传进去。

可以先改成这样：

1. 
   
2. import numpy as np
   
3. import matplotlib.pyplot as plt
   
4. 
   
5. vec_a = np.array([1, 2])
   
6. vec_b = np.array([2, 1])
   
7. 
   
8. vec_sum = vec_a + vec_b
   
9. vec_dif = vec_a - vec_b
   
10. vec_tim = vec_a * 0.5
    
11. 
    
12. print(vec_a)
    
13. print(vec_b)
    
14. print(vec_sum)
    
15. print(vec_dif)
    
16. 
    
17. plt.figure(figsize=(5, 5))
    
18. 
    
19. plt.quiver(0, 0, vec_a[0], vec_a[1],
    
20.            angles='xy', scale_units='xy', scale=1,
    
21.            color='red', label='a')
    
22. 
    
23. plt.quiver(0, 0, vec_b[0], vec_b[1],
    
24.            angles='xy', scale_units='xy', scale=1,
    
25.            color='blue', label='b')
    
26. 
    
27. plt.quiver(0, 0, vec_sum[0], vec_sum[1],
    
28.            angles='xy', scale_units='xy', scale=1,
    
29.            color='lime', label='a+b')
    
30. 
    
31. plt.quiver(0, 0, vec_dif[0], vec_dif[1],
    
32.            angles='xy', scale_units='xy', scale=1,
    
33.            color='violet', label='a-b')
    
34. 
    
35. plt.quiver(0, 0, vec_tim[0], vec_tim[1],
    
36.            angles='xy', scale_units='xy', scale=1,
    
37.            color='yellow', label='0.5a')
    
38. 
    
39. plt.title('Vector Samples')
    
40. plt.xticks(np.arange(-2, 3, 1))
    
41. plt.yticks(np.arange(0, 4, 1))
    
42. plt.legend()
    
43. plt.show()
    

复制代码

记得先调用plt.figure再画图，不然画面尺寸设置不生效。这样就能正确显示向量了。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。