Plotting Complex values with Number of iterations in Python

Question

I am solving Newton ralphson problem in which I am getting complex values of the first derivative, I have to plot the values of the first derivative vs the number of iterations, How do I plot the graph?

I am new to python so I am a little confused.

Edit: I want to plot the real part on the X-axis, the Imaginary on the Y, and the number of iterations on z.

Answer 1

Updated per your comment:

import matplotlib.pyplot as plt

# Define the first_derivative and iterations data
first_derivative = [1 + 1j, 2 + 2j, 3 + 3j, 4 + 4j, 5 + 5j]
iterations = [0, 1, 2, 3, 4]

# Split the first_derivative into real and imaginary parts
x = [val.real for val in first_derivative]
y = [val.imag for val in first_derivative]

# Create a figure and axis for the 3D plot
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the data as a scatter plot
ax.scatter(x, y, iterations, c='r', marker='o')

# Set the x, y, and z labels
ax.set_xlabel('Real Part')
ax.set_ylabel('Imaginary Part')
ax.set_zlabel('Number of Iterations')

# Show the plot
plt.show()

---

Old Code

Using the matplotlib library:

import matplotlib.pyplot as plt

# Example values of first derivative and the number of iterations
first_derivative = [1 + 1j, 2 + 2j, 3 + 3j, 4 + 4j, 5 + 5j]
iterations = [0, 1, 2, 3, 4]

# Plot the values of first derivative vs number of iterations
plt.plot(iterations, first_derivative, 'ro')
plt.xlabel('Number of iterations')
plt.ylabel('First derivative')
plt.title('First derivative vs Number of iterations')
plt.show()

the ro argument passed to the plot function is a format string that specifies the format of the data points to be plotted. The format string consists of two characters:

r : specifies the color of the data points as red .

o : specifies the shape of the data points as circles .

You can use different format strings to specify different colors and shapes for the data points. For example, bx would specify blue crosses , g^ would specify green triangles , and so on.