Swift для TensorFlow включает python совместимость.
https://www.tensorflow.org/swift/tutorials/python_interoperability
Посмотрите, как они просто используют matplotlib:
let np = Python.import("numpy")
let plt = Python.import("matplotlib.pyplot")
let time = np.arange(0, 10, 0.01)
let amplitude = np.exp(-0.1 * time)
let position = amplitude * np.sin(3 * time)
plt.figure(figsize: [15, 10])
plt.plot(time, position)
plt.plot(time, amplitude)
plt.plot(time, -amplitude)
plt.xlabel("Time (s)")
plt.ylabel("Position (m)")
plt.title("Oscillations")
plt.show()
plt.polar
и plt.pie
должны просто работать.