python画轨迹代码问题!

对方的凤飞飞

1. #Python画后向轨迹
   
2. import numpy as np
   
3. import matplotlib.pyplot as plt
   
4. from mpl_toolkits.basemap import Basemap
   
5. import cmaps
   
6. from matplotlib.collections import LineCollection
   
7. 
   
8. 
   
9. from pandas.core.frame import DataFrame
   
10. file_path='1.txt'
    
11. f = open(file_path)
    
12. data = []
    
13. for line in f.readlines():
    
14.     row = []  # 记录每一行
    
15.     lines = line.strip().split("\t")
    
16.     for x in lines:
    
17.         row.append(x)
    
18.     data.append(row)
    
19. f.close()
    
20. print(data[1][9])
    
21. #data is list can not be writen data[1,9]
    
22. newdata=DataFrame(data)#list tranform to dataframe
    
23. fig, ax = plt.subplots()
    
24. 
    
25. #fig=plt.figure(figsize=(25,7))
    
26. m = Basemap(projection='spstere',boundinglat=-60,lon_0=180,resolution='l')
    
27. m.drawcoastlines(linewidth=0.2, color='gray',zorder=3)
    
28. 
    
29. #m.drawparallels(np.arange(-90.,-50.,10.))
    
30. # labels = [left,right,top,bottom]
    
31. #m.drawmeridians(np.arange(-180.,180.,45.))
    
32. parallels = np.arange(-90,-50,10) #纬线
    
33. m.drawparallels(parallels,labels=[False,False,False,False])
    
34. #plt.yticks(parallels,len(parallels)*[''])
    
35. meridians = np.arange(-180,180,45) #经线
    
36. m.drawmeridians(meridians,labels=[False,False,True,True])
    
37. cmap=cmaps.WhiteBlueGreenYellowRed
    
38. 
    
39. 
    
40. lon = newdata[:][2]
    
41. lat = newdata[:][3]
    
42. #h是高度值
    
43. h = newdata[:][4]
    
44. 
    
45. 
    
46. points = np.array([lon, lat]).T.reshape(-1, 1, 2)
    
47. segments = np.concatenate([points[:-1], points[1:]], axis=1)
    
48. print(segments)
    
49. norm = plt.Normalize(0, 1200)
    
50. #设置颜色线条
    
51. lc = LineCollection(segments, cmap=cmap,norm=norm)        
    
52. lc.set_array(h)
    
53. #绘制线条
    
54. line = ax.add_collection(lc)   
    
55. #colorbar
    
56. fig.colorbar(line)#方向
    
57. plt.show()
    
58. plt.savefig('traj.png',dpi=600)

复制代码

1.     resdat -= vmin
   
2. TypeError: unsupported operand type(s) for -: 'str' and 'float'
   

复制代码

上面是报错类型,感觉是我把数据存为数据框类型,变成了series类型,搞了四个小时还没解决!求助!谢谢!

葫芦爷儿

你这报错是数据类型的原因啊，我认为你需要检查txt读取后数据是否为str类型
如果是str，需要转换为float
比如

1. lon = [np.float(x) for x in lon]
   
2. lat = [np.float(x) for x in lat]
   
3. h = [np.float(x) for x in h]

复制代码

另外，你这里为什么要用Line Collection方法，plot不能满足你的需要么？
我觉得你46-47行也需要检查一下，是否和例子的格式相同https://matplotlib.org/gallery/s ... ine_collection.html

对方的凤飞飞

葫芦爷儿 发表于 2020-9-25 21:42
你这报错是数据类型的原因啊，我认为你需要检查txt读取后数据是否为str类型
如果是str，需要转换为float
...

大神說的是呀,請問plot的具體函數是什麼呢?