matplotlibで角度と長さの値を入力して作図したい

Question

red = 15
blue　= 10
angle = 90

上記のような感じで、値を入力することで下のように作図をしたいです。

下のコードで作図することは何とか出来たのですが、簡単にできないかと思い質問させていただきます。
基礎的なことだろうとは思いますがよろしくお願いします。

plt.plot([0,15],[0,0],c='red')
plt.plot([15,(15+(10/2**0.5))],[0,(10/2**0.5)],c='blue')
plt.plot([15,(15+(10/2**0.5))],[0,-(10/2**0.5)],c='blue')

Comments

[aokikenichi]

「簡単にできないか」
とはどういうことでしょうか？
同様の作図を複数するので、逐一書くのが面倒なので関数化したい、というようなことでしょうか。

[yonhark0]

すみません。言葉が足りませんでした。
aokikenichiさんがおっしゃる通り、同様の作図を複数したいと考えていました。
回答もしてくださりありがとうございます！
勉強になります。

[aokikenichi]

解決されたようで良かったです

Answer 1

追記・修正依頼させていただいた想定でよろしければこんなかんじかと

import matplotlib.pyplot as plt
import math

def plot_angle(red, blue, angle):
  plt.plot([0,red],[0,0],c='red')
  plt.plot([red,(red+blue*math.cos(angle/2/180*math.pi))],[0,blue*math.sin(angle/2/180*math.pi)],c='blue')
  plt.plot([red,(red+blue*math.cos(angle/2/180*math.pi))],[0,-blue*math.sin(angle/2/180*math.pi)],c='blue')

plot_angle(15, 10, 90)