Python-从笛卡尔坐标插值到极坐标

气象上，经常要计算轴平均，切向风、径向风，这时需要插值到极坐标后才能计算。这里我们用到scipy模块的griddata插值函数。

首先，导入可能用到的模块

from scipy.interpolate import griddata 
import numpy as np 
import math
import matplotlib.pyplot as plt

 然后，构造原始数据

data = [[1,1,1],[1,0,1],[1,1,1]]
plt.contourf(data)

按照下面代码，进行插值

# # # #构造插值前的格点矩阵
x_or = np.arange(3)  #原始坐标
y_or = np.arange(3)
x_d,y_d = np.meshgrid(x_or,y_or)
#
#构造所需插值的极坐标格点
theta = np.linspace(0,360,361)   #角度
r = np.linspace(0,1,2)           #半径
cx = 1                           #极坐标中心位置，可更改
cy = 1                           #极坐标中心位置，可更改
#
#将极坐标格点转换到笛卡尔坐标下
xcoord = np.zeros((2,361))
ycoord = np.zeros((2,361))