for文での繰り返しをmapで

Question

import matplotlib.pyplot as plt

a = 1200
b = 1800
L = 3000

print("E")
E = input()
E = int(E)

print("I")
I = input()
I = int(I)

print("P")
P = input()
P = int(P)

print("E2")
E2 = input()
E2 = int(E2)

print("I2")
I2 = input()
I2 = int(I2)

print("P2")
P2=input()
P2=int(P2)

def f(x_i) :
  if x_i < a:
    return (b * P) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E * I * L)
  else:
    return (a * P) / (6 * E * I * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))

  if x_i < a:
    return (b * P2) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E2 * I2 * L)
  else:
    return (a * P2) / (6 * E2 * I2 * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))

x = list(range(L))
y=list(map(f,x))
y2=list(map(f,x))

plt.grid(which="major", color="gray", linestyle="-")
plt.xlim(0, L)
plt.ylim(min(y) - 1, max(y) + 1)



plt.plot(x, y, label = "label-A")
plt.plot(x,y2, label = "label-B")
plt.legend(loc = 'lower right')
plt.show()
```### 前提・実現したいこと

for文をmap文にして繰り返しをおこないたい


### 該当のソースコード

```Python
import matplotlib.pyplot as plt

a = 1200
b = 1800
L = 3000

print("E")
E = input()
E = int(E)

print("I")
I = input()
I = int(I)

print("P")
P = input()
P = int(P)

print("E2")
E2 = input()
E2 = int(E2)

print("I2")
I2 = input()
I2 = int(I2)

print("P2")
P2=input()
P2=int(P2)


x = list(range(L))
y = [(b * P) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E * I * L)
     if x_i < a else (a * P) / (6 * E * I * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i)) for x_i in x]

y2 = [(b * P2) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E2 * I2 * L)
     if x_i < a else (a * P2) / (6 * E2 * I2 * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i)) for x_i in x]

plt.grid(which="major", color="gray", linestyle="-")
plt.xlim(0, L)
plt.ylim(min(y) - 1, max(y) + 1)



plt.plot(x, y, label = "label-A")
plt.plot(x,y2, label = "label-B")
plt.legend(loc = 'lower right')
plt.show()

Answer 1

実際に動作はさせていませんが、以下でよいと思います。

def f(x_i):
    if x_i < a:
        return (b * P) * ～
    else:
        return (a * P) / ～

x = list(range(L))
y = list(map(f, x))

Comments

[shion_shion]

実際に動作させて見た所
NameError: name 'x_1' is not defined
このようなエラーが出てしまいます。
プログラミングを始めたばかりで分からない事が多いのですがどのようにすればよろしいでしょうか？

[can110]

元のコードが分かりませんが「x_1」は「x_i」のタイプミスでは？

[shion_shion]

すみません、そこはご指摘の通りでした。
しかし、実際に実行し、グラフを表示した所"label-A"のグラフが表示されなくなってしまったのですが、原因わかりますでしょうか？

[can110]

コードが分からないので何とも言えませんが、修正前後でスペルやコードミスがあるのではないでしょうか。

[shion_shion]

すみません、定義を1つし忘れていたようです。
解決いたしました。ありがとうございました。